Spaces:
Sleeping
Sleeping
Gaurav vashistha commited on
Commit ·
e864198
1
Parent(s): 016d4bc
Bugfix: Remove dynamic models/ string formatting in Vision Agent
Browse files- agents/memory_agent.py +2 -2
agents/memory_agent.py
CHANGED
|
@@ -28,7 +28,7 @@ class MemoryAgent:
|
|
| 28 |
print(f"Creating index {self.index_name}...")
|
| 29 |
self.pc.create_index(
|
| 30 |
name=self.index_name,
|
| 31 |
-
dimension=768, #
|
| 32 |
metric='cosine',
|
| 33 |
spec=ServerlessSpec(
|
| 34 |
cloud='aws',
|
|
@@ -43,7 +43,7 @@ class MemoryAgent:
|
|
| 43 |
self.index = self.pc.Index(self.index_name)
|
| 44 |
|
| 45 |
def _get_embedding(self, text):
|
| 46 |
-
# Using
|
| 47 |
result = self.client.models.embed_content(
|
| 48 |
model="gemini-embedding-001",
|
| 49 |
contents=text,
|
|
|
|
| 28 |
print(f"Creating index {self.index_name}...")
|
| 29 |
self.pc.create_index(
|
| 30 |
name=self.index_name,
|
| 31 |
+
dimension=768, # gemini-embedding-001 output dimension
|
| 32 |
metric='cosine',
|
| 33 |
spec=ServerlessSpec(
|
| 34 |
cloud='aws',
|
|
|
|
| 43 |
self.index = self.pc.Index(self.index_name)
|
| 44 |
|
| 45 |
def _get_embedding(self, text):
|
| 46 |
+
# Using gemini-embedding-001
|
| 47 |
result = self.client.models.embed_content(
|
| 48 |
model="gemini-embedding-001",
|
| 49 |
contents=text,
|