Merge pull request #1020 from julep-ai/d/coobooks

D/cookbooks: Added new cookbooks
julep-ai · Jan 6, 2025 · 74817a3 · 74817a3
2 parents e68ccf0 + 0819e0e
commit 74817a3
Show file tree

Hide file tree

Showing 11 changed files with 7,856 additions and 3,592 deletions.
diff --git a/agents-api/notebooks/01-revise-entities.ipynb b/agents-api/notebooks/01-revise-entities.ipynb
diff --git a/agents-api/notebooks/02-trim-messages.ipynb b/agents-api/notebooks/02-trim-messages.ipynb
diff --git a/agents-api/notebooks/03-summarise.ipynb b/agents-api/notebooks/03-summarise.ipynb
diff --git a/agents-api/notebooks/RecSum-experiments.ipynb b/agents-api/notebooks/RecSum-experiments.ipynb
diff --git a/agents-api/notebooks/test-chat.json b/agents-api/notebooks/test-chat.json
diff --git a/agents-api/pyproject.toml b/agents-api/pyproject.toml
@@ -23,7 +23,7 @@ dependencies = [
   "jsonschema~=4.22.0",
   "langchain-core~=0.3.14",
   "larch-pickle~=1.4.3",
-  "litellm",
+  "litellm~1.57.0",
   "lz4~=4.3.3",
   "msgpack~=1.1.0",
   "numpy>=2.0.0,<2.1.0",
@@ -77,6 +77,6 @@ py-modules = [
     "agents_api",
 ]
 
-[tool.uv.sources]
-litellm = { url = "https://github.com/julep-ai/litellm/archive/fix_anthropic_tool_image_content.zip" }
+#[tool.uv.sources]
+#litellm = { url = "https://github.com/julep-ai/litellm/archive/fix_anthropic_tool_image_content.zip" }
 
diff --git a/agents-api/tests/test_docs_queries.py b/agents-api/tests/test_docs_queries.py
@@ -272,7 +272,7 @@ async def _(
     result = await search_docs_by_embedding(
         developer_id=developer.id,
         owners=[("agent", agent.id)],
-        query_embedding=query_embedding,
+        embedding=query_embedding,
         k=3,  # Add k parameter
         metadata_filter={"test": "test"},  # Add metadata filter
         connection_pool=pool,