fix bugs

agents/legacy/jar3d_agent.py

@@ -279,6 +279,7 @@ class NoToolExpert(BaseAgent[State]):
 
     def get_prompt(self, state) -> str:
         # print(f"\nn{state}\n")
+        # The prompt not from the markdown, but form the meta-expert generated
         system_prompt = state["meta_prompt"][-1].content
         return system_prompt
         

models/llms.py

@@ -32,8 +32,8 @@ class BaseModel:
 class MistralModel(BaseModel):
     def __init__(self, temperature: float, model: str, json_response: bool, max_retries: int = 3, retry_delay: int = 1):
         super().__init__(temperature, model, json_response, max_retries, retry_delay)
-        config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-        load_config(config_path)
+        # config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+        # load_config(config_path)
         # load_config()
         self.api_key = os.environ.get("MISTRAL_API_KEY")
         self.headers = {
@@ -94,8 +94,8 @@ class MistralModel(BaseModel):
 class ClaudeModel(BaseModel):
     def __init__(self, temperature: float, model: str, json_response: bool, max_retries: int = 3, retry_delay: int = 1):
         super().__init__(temperature, model, json_response, max_retries, retry_delay)
-        config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-        load_config(config_path)
+        # config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+        # load_config(config_path)
         self.api_key = os.environ.get("ANTHROPIC_API_KEY")
         self.headers = {
             'Content-Type': 'application/json', 
@@ -147,8 +147,8 @@ class ClaudeModel(BaseModel):
 class GeminiModel(BaseModel):
     def __init__(self, temperature: float, model: str, json_response: bool, max_retries: int = 3, retry_delay: int = 1):
         super().__init__(temperature, model, json_response, max_retries, retry_delay)
-        config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-        load_config(config_path)
+        # config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+        # load_config(config_path)
         self.api_key = os.environ.get("GEMINI_API_KEY")
         self.headers = {
             'Content-Type': 'application/json'
@@ -218,8 +218,8 @@ class GeminiModel(BaseModel):
 class GroqModel(BaseModel):
     def __init__(self, temperature: float, model: str, json_response: bool, max_retries: int = 3, retry_delay: int = 1):
         super().__init__(temperature, model, json_response, max_retries, retry_delay)
-        config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-        load_config(config_path)
+        # config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+        # load_config(config_path)
         self.api_key = os.environ.get("GROQ_API_KEY")
         self.headers = {
             'Content-Type': 'application/json', 

tools/legacy/offline_graph_rag_tool copy.py

@@ -25,12 +25,14 @@ from langchain.schema import Document
 from config.load_configs import load_config
 from langchain_community.docstore.in_memory import InMemoryDocstore
 from fake_useragent import UserAgent
+from dotenv import load_dotenv
 
 root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.insert(0, root_dir)
 
-config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-load_config(config_path)
+# config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+# load_config(config_path)
+load_dotenv()
 
 ua = UserAgent()
 os.environ["USER_AGENT"] = ua.random

tools/offline_graph_rag_tool.py

@@ -21,12 +21,14 @@ from langchain.schema import Document
 from config.load_configs import load_config
 from langchain_community.docstore.in_memory import InMemoryDocstore
 from fake_useragent import UserAgent
-
+from dotenv import load_dotenv
 root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.insert(0, root_dir)
 
-config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
-load_config(config_path)
+# config_path = os.path.join(os.path.dirname(__file__), '..', 'config', 'config.yaml')
+# load_config(config_path)
+
+load_dotenv()
 
 ua = UserAgent()
 os.environ["USER_AGENT"] = ua.random