Gaurav vashistha commited on
Commit
e864198
·
1 Parent(s): 016d4bc

Bugfix: Remove dynamic models/ string formatting in Vision Agent

Browse files
Files changed (1) hide show
  1. 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, # models/text-embedding-004 dimension
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 models/text-embedding-004
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,