trying to make tests work

longevity-genie · Jan 19, 2025 · fe03e58 · fe03e58
1 parent 2f62d97
commit fe03e58
Show file tree

Hide file tree

Showing 7 changed files with 47 additions and 3 deletions.
diff --git a/docs/__init__.py b/docs/__init__.py
@@ -0,0 +1 @@
+"""Documentation package for just-agents.""" 
diff --git a/docs/api.rst b/docs/api.rst
@@ -0,0 +1,7 @@
+API Reference
+============
+
+.. automodule:: just_agents
+   :members:
+   :undoc-members:
+   :show-inheritance: 
diff --git a/docs/conf.py b/docs/conf.py
@@ -9,7 +9,18 @@
     'sphinx.ext.autodoc',
     'sphinx.ext.napoleon',
     'sphinx.ext.viewcode',
-    'myst_parser'
+    'myst_parser',
+    'sphinx.ext.intersphinx'
+]
+
+# Add after the existing extensions
+autodoc_mock_imports = [
+    'litellm',
+    'pydantic',
+    'requests',
+    'numpydoc',
+    'rich',
+    'just_agents'
 ]
 
 # Theme settings
@@ -19,4 +30,19 @@
 source_suffix = {
     '.rst': 'restructuredtext',
     '.md': 'markdown',
+}
+
+# Add any paths that contain templates here, relative to this directory
+templates_path = ['_templates']
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
+
+# The master toctree document
+master_doc = 'index'
+
+# Intersphinx mapping
+intersphinx_mapping = {
+    'python': ('https://docs.python.org/3', None),
 }
diff --git a/docs/examples.rst b/docs/examples.rst
@@ -0,0 +1,7 @@
+Examples
+========
+
+Here you'll find various examples of how to use just-agents.
+
+.. toctree::
+   :maxdepth: 2
diff --git a/docs/index.rst b/docs/index.rst
@@ -6,6 +6,8 @@ Welcome to just-agents documentation!
    :caption: Contents:
 
    readme
+   api
+   examples
 
 Indices and tables
 ==================

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,3 +1,4 @@
 sphinx>=7.0.0
 sphinx-rtd-theme>=1.3.0
-myst-parser>=2.0.0
+myst-parser>=2.0.0
+sphinx-autodoc-typehints>=1.25.2
diff --git a/tests/test_chain_of_thought.py b/tests/test_chain_of_thought.py
@@ -15,7 +15,7 @@ def count_letters(character: str, word: str) -> str:
 def test_function_query():
     load_dotenv(override = True)
 
-    llm_options = just_agents.llm_options.LLAMA3_3
+    llm_options = just_agents.llm_options.OPENAI_O1_MINI
     agent: ChainOfThoughtAgent = ChainOfThoughtAgent(llm_options=llm_options, tools=[count_letters])
     result, thoughts = agent.think("Count the number of occurrences of the letter 'L' in the word - 'LOLLAPALOOZA'.")
     print("Thoughts: ", thoughts)