Merge pull request #1026 from julep-ai/x/migration-fix

X/migration fix: Fixed migrations and docs search route

agents-api/agents_api/queries/docs/search_docs_by_embedding.py

@@ -36,7 +36,7 @@ async def search_docs_by_embedding(
     embedding: list[float],
     k: int = 10,
     owners: list[tuple[Literal["user", "agent"], UUID]],
-    confidence: float = 0.5,
+    confidence: int | float = 0.5,
     metadata_filter: dict[str, Any] = {},
 ) -> tuple[str, list]:
     """

agents-api/agents_api/queries/docs/search_docs_hybrid.py

@@ -46,7 +46,7 @@ async def search_docs_hybrid(
     alpha: float = 0.5,
     metadata_filter: dict[str, Any] = {},
     search_language: str = "english",
-    confidence: float = 0.5,
+    confidence: int | float = 0.5,
 ) -> tuple[str, list]:
     """
     Hybrid text-and-embedding doc search. We get top-K from each approach,

agents-api/tests/test_docs_routes.py

@@ -69,7 +69,7 @@ async def _(make_request=make_request, agent=test_agent):
         assert response.status_code == 200
         assert response.json()["id"] == doc_id
         assert response.json()["title"] == "Test Agent Doc"
-        assert response.json()["content"] == "This is a test agent document."
+        assert response.json()["content"] == ["This is a test agent document."]
 
         response = make_request(
             method="DELETE",

memory-store/migrations/000021_fix_toolname_contraint.down.sql

@@ -0,0 +1,10 @@
+BEGIN;
+
+-- Drop the updated unique constraint
+ALTER TABLE tools DROP CONSTRAINT IF EXISTS ct_unique_name_per_agent;
+
+-- Restore the original unique constraint (without developer_id)
+ALTER TABLE tools ADD CONSTRAINT ct_unique_name_per_agent 
+UNIQUE (agent_id, name, task_id);
+
+COMMIT;

memory-store/migrations/000021_fix_toolname_contraint.up.sql

@@ -0,0 +1,10 @@
+BEGIN;
+
+-- Drop the existing unique constraint
+ALTER TABLE tools DROP CONSTRAINT IF EXISTS ct_unique_name_per_agent;
+
+-- Add the new unique constraint including developer_id
+ALTER TABLE tools ADD CONSTRAINT ct_unique_name_per_agent 
+UNIQUE (developer_id, agent_id, name, task_id);
+
+COMMIT;