fix(agents-api): Fix tests related to executions

Signed-off-by: Diwank Singh Tomer <diwank.singh@gmail.com>

agents-api/agents_api/common/storage_handler.py

@@ -15,6 +15,9 @@
 
 
 def store_in_blob_store_if_large(x: Any) -> RemoteObject | Any:
+    if not use_blob_store_for_temporal:
+        return x
+
     s3.setup()
 
     serialized = serialize(x)
@@ -28,6 +31,9 @@ def store_in_blob_store_if_large(x: Any) -> RemoteObject | Any:
 
 
 def load_from_blob_store_if_remote(x: Any | RemoteObject) -> Any:
+    if not use_blob_store_for_temporal:
+        return x
+
     s3.setup()
 
     if isinstance(x, RemoteObject):
@@ -143,4 +149,4 @@ async def wrapper(*args, **kwargs) -> Any:
 
         return result
 
-    return wrapper
+    return wrapper if use_blob_store_for_temporal else f

agents-api/agents_api/env.py

@@ -118,6 +118,12 @@
     temporal_worker_url=temporal_worker_url,
     temporal_namespace=temporal_namespace,
     embedding_model_id=embedding_model_id,
+    use_blob_store_for_temporal=use_blob_store_for_temporal,
+    blob_store_bucket=blob_store_bucket,
+    blob_store_cutoff_kb=blob_store_cutoff_kb,
+    s3_endpoint=s3_endpoint,
+    s3_access_key=s3_access_key,
+    s3_secret_key=s3_secret_key,
     testing=testing,
 )
 

agents-api/agents_api/models/docs/search_docs_by_text.py

@@ -63,7 +63,6 @@ def search_docs_by_text(
         [owner_type, str(owner_id)] for owner_type, owner_id in owners
     ]
 
-    # Need to use NEAR/3($query) to search for arbitrary text within 3 words of each other
     # See: https://docs.cozodb.org/en/latest/vector.html#full-text-search-fts
     fts_queries = paragraph_to_custom_queries(query) or [
         re.sub(r"[^\w\s\-_]+", "", query)