2

.gitignore

@@ -1,4 +1,5 @@
 /config/config.yaml
+.env
 __pycache__/
 *.pyc
 /.vscode

README.md

@@ -26,6 +26,10 @@ Thanks John Adeojo, who brings this wonderful project to open source community!
    - Hugging Face deploy
 
 ## TODO
+[] fix local docker bugs,local test
+[] deploy to Huggingface
+
+
 [] Long-term memory.
 [] Full Ollama and vLLM integration.
 [] Integrations to RAG platforms for more intelligent document processing and faster RAG.

agents/meta_agent.py

@@ -473,7 +473,7 @@ if __name__ == "__main__":
             break
 
         # current_time = datetime.now()
-        recursion_limit = 15
+        recursion_limit = 30
         state["recursion_limit"] = recursion_limit
         state["user_input"] = query
         limit = {"recursion_limit": recursion_limit}

chat.py

@@ -19,10 +19,11 @@ from agents.jar3d_agent import (State,
 from agents.base_agent import BaseAgent
 from utils.read_markdown import read_markdown_file
 from config.load_configs import load_config
+from dotenv import load_dotenv
 
 
-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 environment variables from .env file
 load_dotenv()

models/llms.py

@@ -7,6 +7,8 @@ from typing import List, Dict
 from utils.logging import log_function, setup_logging
 from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
 from config.load_configs import load_config
+from dotenv import load_dotenv
+
 
 setup_logging(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
@@ -30,6 +32,7 @@ class MistralModel(BaseModel):
         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)
+        # load_config()
         self.api_key = os.environ.get("MISTRAL_API_KEY")
         self.headers = {
             'Content-Type': 'application/json',
@@ -389,8 +392,9 @@ class VllmModel(BaseModel):
 class OpenAIModel(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_dotenv()
         self.model_endpoint = 'https://api.302.ai/v1/chat/completions'
         self.api_key = os.getenv('OPENAI_API_KEY')
         self.headers = {

tools/google_serper.py

@@ -6,6 +6,7 @@ sys.path.insert(0, root_dir)
 import requests
 from typing import Dict, Any
 from config.load_configs import load_config
+from dotenv import load_dotenv
 
 def format_results(organic_results: str) -> str:
     result_strings = []
@@ -33,8 +34,9 @@ def format_shopping_results(shopping_results: list) -> str:
     return '\n'.join(result_strings)
 
 def serper_search(query: str, location: str) -> Dict[str, Any]:
-    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()
     search_url = "https://google.serper.dev/search"
     headers = {
         'Content-Type': 'application/json',
@@ -63,8 +65,9 @@ def serper_search(query: str, location: str) -> Dict[str, Any]:
         return f"JSON decoding error occurred: {json_err}"
     
 def serper_shopping_search(query: str, location: str) -> Dict[str, Any]:
-    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()
     search_url = "https://google.serper.dev/shopping"
     headers = {
         'Content-Type': 'application/json',
@@ -89,8 +92,9 @@ def serper_shopping_search(query: str, location: str) -> Dict[str, Any]:
         return f"JSON decoding error occurred: {json_err}"
 
 def serper_scholar_search(query: str, location: str) -> Dict[str, Any]:
-    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()
     search_url = "https://google.serper.dev/scholar"
     headers = {
         'Content-Type': 'application/json',

tools/legacy/offline_rag_tool.py

@@ -24,13 +24,15 @@ from config.load_configs import load_config
 from langchain_community.docstore.in_memory import InMemoryDocstore
 from fake_useragent import UserAgent
 from multiprocessing import Pool, cpu_count
+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
 os.environ["FAISS_OPT_LEVEL"] = "generic"

tools/legacy/rag_tool.py

@@ -23,13 +23,15 @@ 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
 os.environ["FAISS_OPT_LEVEL"] = "generic"

tools/offline_graph_rag_tool_with_async.py

@@ -26,12 +26,14 @@ from config.load_configs import load_config
 from langchain_community.docstore.in_memory import InMemoryDocstore
 from fake_useragent import UserAgent
 import asyncio
+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