bug fix

VRMKX · Nov 24, 2023 · b61f10e · b61f10e
1 parent cb74875
commit b61f10e
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 13 deletions.
diff --git a/ai/agents/code_utils.py b/ai/agents/code_utils.py
@@ -8,10 +8,10 @@
 import time
 from hashlib import md5
 import logging
-from agents import oai
+from ai.agents import oai
 import json
 import traceback
-from backend.base_config import if_hide_sensitive
+from ai.backend.base_config import if_hide_sensitive
 
 try:
     import docker
@@ -607,7 +607,7 @@ async def tell_logger(websocket, log_str):
                                ******* python code ************
                                ******* python code ************
                                ******* python code ************
-    
+
                                ```
                                    """
                 log_str = re.sub(CODE_BLOCK_PATTERN, replacement, log_str, flags=re.DOTALL)

diff --git a/ai/agents/oai/__init__.py b/ai/agents/oai/__init__.py
@@ -1,5 +1,5 @@
-from agents.oai.completion import Completion, ChatCompletion
-from agents.oai.openai_utils import (
+from ai.agents.oai.completion import Completion, ChatCompletion
+from ai.agents.oai.openai_utils import (
     get_config_list,
     config_list_gpt4_gpt35,
     config_list_openai_aoai,

diff --git a/ai/backend/chat_task.py b/ai/backend/chat_task.py
@@ -2,13 +2,13 @@
 import json
 import time
 import traceback
-from backend.util.write_log import logger
-from agents import AgentInstanceUtil
-from backend.memory import ChatMemoryManager
-from backend.base_config import CONFIG
-from backend.aidb.report import ReportMysql, ReportPostgresql
-from backend.aidb.analysis import AnalysisMysql, AnalysisCsv, AnalysisPostgresql
-from backend.aidb import AIDB
+from ai.backend.util.write_log import logger
+from ai.agents import AgentInstanceUtil
+from ai.backend.memory import ChatMemoryManager
+from ai.backend.base_config import CONFIG
+from ai.backend.aidb.report import ReportMysql, ReportPostgresql
+from ai.backend.aidb.analysis import AnalysisMysql, AnalysisCsv, AnalysisPostgresql
+from ai.backend.aidb import AIDB
 
 message_pool: ChatMemoryManager = ChatMemoryManager(name="message_pool")
 

diff --git a/ai/main.py b/ai/main.py
@@ -1,5 +1,5 @@
 import asyncio
-from backend.start_server import WSServer
+from ai.backend.start_server import WSServer
 
 if __name__ == '__main__':
     server_port = 8339