feat(agents-api): Move blob storage to cozo

agents-api/.gitignore

@@ -1,5 +1,6 @@
 # Local database files
-cozo.db
+cozo*
+.cozo*
 temporal.db
 *.bak
 *.dat

agents-api/agents_api/activities/demo.py

@@ -1,5 +1,3 @@
-from typing import Callable
-
 from temporalio import activity
 
 from ..env import testing
@@ -14,6 +12,6 @@ async def mock_demo_activity(a: int, b: int) -> int:
     return a + b
 
 
-demo_activity: Callable[[int, int], int] = activity.defn(name="demo_activity")(
+demo_activity = activity.defn(name="demo_activity")(
     demo_activity if not testing else mock_demo_activity
 )

agents-api/agents_api/activities/truncation.py

@@ -14,10 +14,10 @@
 def get_extra_entries(messages: list[Entry], token_count_threshold: int) -> list[UUID]:
     raise NotImplementedError()
 
-    if not len(messages):
-        return messages
+    # if not len(messages):
+    #     return messages
 
-    _token_cnt, _offset = 0, 0
+    # _token_cnt, _offset = 0, 0
     # if messages[0].role == Role.system:
     #     token_cnt, offset = messages[0].token_count, 1
 
@@ -36,7 +36,8 @@ def get_extra_entries(messages: list[Entry], token_count_threshold: int) -> list
 @activity.defn
 @beartype
 async def truncation(session_id: str, token_count_threshold: int) -> None:
-    session_id = UUID(session_id)
+    raise NotImplementedError()
+    # session_id = UUID(session_id)
 
     # delete_entries(
     #     get_extra_entries(

agents-api/pyproject.toml

@@ -60,6 +60,7 @@ dev = [
     "ipywidgets>=8.1.5",
     "julep>=1.43.1",
     "jupyterlab>=4.3.1",
+    "pip>=24.3.1",
     "poethepoet>=0.31.1",
     "pyjwt>=2.10.1",
     "pyright>=1.1.389",
@@ -68,6 +69,11 @@ dev = [
     "ward>=0.68.0b0",
 ]
 
+[tool.setuptools]
+py-modules = [
+    "agents_api"
+]
+
 [tool.uv.sources]
 litellm = { url = "https://github.com/julep-ai/litellm/archive/fix_anthropic_tool_image_content.zip" }