improvements with tests and sqlite functions

.github/workflows/run_tests.yaml

@@ -16,6 +16,10 @@ jobs:
 
     env:
       GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }}
+      GROQ_API_KEY_1: ${{ secrets.GROQ_API_KEY_1 }}
+      GROQ_API_KEY_2: ${{ secrets.GROQ_API_KEY_2 }}
+      GROQ_API_KEY_3: ${{ secrets.GROQ_API_KEY_3 }}
+      GROQ_API_KEY_4: ${{ secrets.GROQ_API_KEY_4 }}
       OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
 
     steps:

.gitignore

@@ -168,4 +168,5 @@ cython_debug/
 
 *.egg-info
 
-config/agent_profiles.yaml #to deal with testing bug
+./config/ #temporal config produced by tests
+config/agent_profiles.yaml #to deal with testing bug

.readthedocs.yaml

@@ -6,6 +6,8 @@ build:
   tools:
     python: "3.11"
   commands:
+    - pip install poetry
+    # First modify all pyproject.toml files to use exact versions
     - cp pyproject.toml pyproject.toml.bak
     - |
       sed -i \
@@ -16,12 +18,25 @@ build:
         -e "s|{ path = \"router\", develop = true }|\"0.4.9\"|g" \
         -e "s|{ path = \"examples\", develop = true }|\"0.4.9\"|g" \
         pyproject.toml
+    # Build packages in order following build.sh approach
+    # First build core
+    - cd core && poetry build && cd ..
+    - pip install core/dist/*.whl
+    # Build and install other packages
+    - cd tools && poetry build && cd .. && pip install tools/dist/*.whl
+    - cd coding && poetry build && cd .. && pip install coding/dist/*.whl
+    - cd web && poetry build && cd .. && pip install web/dist/*.whl
+    - cd router && poetry build && cd .. && pip install router/dist/*.whl
+    - cd examples && poetry build && cd .. && pip install examples/dist/*.whl
+    # Build the meta-package
+    - touch ./just_agents/__init__.py
+    - poetry build
+    - pip install dist/*.whl
+    - rm ./just_agents/__init__.py
 
 sphinx:
   configuration: docs/conf.py
 
 python:
   install:
-    - method: pip
-      path: .
     - requirements: docs/requirements.txt

docs/conf.py

@@ -11,7 +11,8 @@
     'sphinx.ext.viewcode',
     'myst_parser',
     'sphinx.ext.intersphinx',
-    'sphinx_autodoc_typehints'
+    'sphinx_autodoc_typehints',
+    'sphinx.ext.autoapi'
 ]
 
 # Add after the existing extensions
@@ -20,8 +21,7 @@
     'pydantic',
     'requests',
     'numpydoc',
-    'rich',
-    'just_agents'
+    'rich'
 ]
 
 # Theme settings
@@ -52,4 +52,10 @@
 myst_enable_extensions = [
     "colon_fence",
     "deflist"
-]
+]
+
+# AutoAPI settings
+autoapi_type = 'python'
+autoapi_dirs = ['../']
+autoapi_options = ['members', 'undoc-members', 'show-inheritance', 'show-module-summary']
+autoapi_python_use_implicit_namespaces = True

tests/test_chat_agent.py

@@ -53,7 +53,9 @@ def test_quering(open_genes_db):
                     goal=f"help users by using SQL syntax to form comands to work with the {open_genes_db} sqlite database",
                     task="formulate appropriate comands to operate in the given database.",
                     tools=[sqlite_query],
-                    llm_options=LLAMA3_3)
+                    llm_options=LLAMA3_3,
+                    key_list_env="GROQ_API_KEY" # LOAD GROQ_API_KEY FROM ENVIRONMENT VARIABLES
+                    )
 
 
     response = agent.query("Show me all tables in the database")

tests/test_pure_litellm.py

@@ -3,8 +3,6 @@
 import pytest
 from dotenv import load_dotenv
 
-
-
 def get_current_weather(location: str):
     """
     Gets the current weather in a given location
@@ -58,7 +56,7 @@ def test_oai_works(load_env):
     OPENAI_GPT4oMINI,_ = load_env
     execute_completion(OPENAI_GPT4oMINI)
 
-@pytest.mark.skip(reason="until fixed in https://github.com/BerriAI/litellm/issues/7621")
+#@pytest.mark.skip(reason="until fixed in https://github.com/BerriAI/litellm/issues/7621")
 def test_grok_bug(load_env):
     _, LLAMA3_3 = load_env
     execute_completion(LLAMA3_3)

tests/test_secretary_agent.py

@@ -4,16 +4,18 @@
 import json
 import just_agents.llm_options
 from just_agents.router.secretary_agent import SecretaryAgent
+from pytest import TempPathFactory, FixtureRequest
+from typing import Tuple, Any
 
 @pytest.fixture(scope='module')
-def temp_config_path(tmpdir_factory):
+def temp_config_path(tmpdir_factory: TempPathFactory) -> str:
     # Create a temporary directory for YAML files
     dotenv.load_dotenv(override=True)
     tmpdir_factory.mktemp('config')
     return SecretaryAgent.DEFAULT_CONFIG_PATH
 
 @pytest.fixture
-def secretary_autoload_false(temp_config_path):
+def secretary_autoload_false(temp_config_path: str) -> Tuple[SecretaryAgent, bool]:
     dotenv.load_dotenv(override=True)
     params = {
         'autoload_from_yaml': False,
@@ -28,7 +30,7 @@ def secretary_autoload_false(temp_config_path):
     return secretary, result
 
 @pytest.fixture
-def secretary_autoload_true(temp_config_path, secretary_autoload_false):
+def secretary_autoload_true(temp_config_path: str, secretary_autoload_false: Tuple[SecretaryAgent, bool]) -> Tuple[SecretaryAgent, bool]:
     # Ensure the YAML file exists from the previous fixture
     dotenv.load_dotenv(override=True)
     params = {
@@ -47,22 +49,22 @@ def secretary_autoload_true(temp_config_path, secretary_autoload_false):
     secretary.save_to_yaml()
     return secretary, result
 
-def test_secretary_autoload_false(secretary_autoload_false):
+def test_secretary_autoload_false(secretary_autoload_false: Tuple[SecretaryAgent, bool]) -> None:
     dotenv.load_dotenv(override=True)
     secretary, result = secretary_autoload_false
     assert result is True, "Failed to update profile when autoload_from_yaml is False."
     assert secretary.description != secretary.DEFAULT_DESCRIPTION, "Description was not updated."
     assert secretary.llm_model_name is not None, "LLM model name is None."
 
-def test_secretary_autoload_true(secretary_autoload_true):
+def test_secretary_autoload_true(secretary_autoload_true: Tuple[SecretaryAgent, bool]) -> None:
     dotenv.load_dotenv(override=True)
     secretary, result = secretary_autoload_true
     assert result is True, "Failed to update profile when autoload_from_yaml is True."
     assert secretary.description != secretary.DEFAULT_DESCRIPTION, "Description was not updated."
     assert secretary.extras.get("personality_traits") is not None, "Personality traits were not set."
     assert secretary.llm_model_name is not None, "LLM model name is None."
 
-def test_new_secretary(temp_config_path):
+def test_new_secretary(temp_config_path: str) -> None:
     # Load the secretary from the YAML file created in previous tests
     dotenv.load_dotenv(override=True)
     new_secretary = SecretaryAgent.from_yaml(

tests/test_stream.py

@@ -5,6 +5,8 @@
 from just_agents.base_agent import BaseAgent
 from just_agents.llm_options import LLMOptions, LLAMA3_3, LLAMA3_2_VISION, OPENAI_GPT4oMINI
 from just_agents.just_tool import JustToolsBus
+
+
 @pytest.fixture(scope="module", autouse=True)
 def load_env():
     load_dotenv(override=True)