fix(agents-api): chnaged from voyage to openai

julep-ai · Jan 8, 2025 · 635b227 · 635b227
1 parent 94ace86
commit 635b227
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/agents-api/agents_api/queries/docs/create_doc.py b/agents-api/agents_api/queries/docs/create_doc.py
@@ -69,7 +69,7 @@ async def create_doc(
     owner_type: Literal["user", "agent"],
     owner_id: UUID,
     modality: Literal["text", "image", "mixed"] | None = "text",
-    embedding_model: str | None = "voyage-3",
+    embedding_model: str | None = "text-embedding-3-large",
     embedding_dimensions: int | None = 1024,
     language: str | None = "english",
     index: int | None = 0,

diff --git a/agents-api/tests/utils.py b/agents-api/tests/utils.py
@@ -132,8 +132,8 @@ def get_pg_dsn(start_vectorizer: bool = False):
                 pg_dsn.replace("localhost", postgres.get_container_host_ip()),
             )
             .with_env(
-                "VOYAGE_API_KEY",
-                os.environ.get("VOYAGE_API_KEY"),
+                "OPENAI_API_KEY",
+                os.environ.get("OPENAI_API_KEY"),
             )
         ) as vectorizer:
             wait_for_logs(

diff --git a/memory-store/docker-compose.yml b/memory-store/docker-compose.yml
@@ -7,7 +7,7 @@ services:
     # See: https://github.com/timescale/timescaledb-docker?tab=readme-ov-file#notes-on-timescaledb-tune
     environment:
       - POSTGRES_PASSWORD=${MEMORY_STORE_PASSWORD:-postgres}
-      - VOYAGE_API_KEY=${VOYAGE_API_KEY}
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
     ports:
       - "5432:5432"
     volumes:
@@ -29,8 +29,8 @@ services:
     image: timescale/pgai-vectorizer-worker:v0.3.0
     environment:
       - PGAI_VECTORIZER_WORKER_DB_URL=postgres://postgres:${MEMORY_STORE_PASSWORD:-postgres}@memory-store:5432/postgres
-      - VOYAGE_API_KEY=${VOYAGE_API_KEY}
-    command: [ "--poll-interval", "5s" ]
+      - OPENAI_API_KEY=${OPENAI_API_KEY}
+    command: [ "--poll-interval", "2s" ]
     depends_on:
       memory-store:
         condition: service_healthy