helloparthshah commited on
Commit
3e394af
·
1 Parent(s): 093c145

fixed memories

Browse files
Files changed (1) hide show
  1. src/manager/manager.py +15 -6
src/manager/manager.py CHANGED
@@ -172,18 +172,28 @@ class GeminiManager:
172
  if isinstance(message["content"], tuple):
173
  path = message["content"][0]
174
  try:
175
- file = self.client.files.upload(file=path)
176
- formatted_history.append(file)
 
 
 
 
 
177
  except Exception as e:
178
  logger.error(f"Error uploading file: {e}")
179
- formatted_history.append(
180
- types.Part.from_text(text="Error uploading file: "+str(e)))
 
 
 
 
 
181
  continue
182
  else:
183
  parts = [types.Part.from_text(
184
  text=message.get("content", ""))]
185
  case "memories":
186
- role = "user"
187
  parts = [types.Part.from_text(
188
  text="Relevant memories: "+message.get("content", ""))]
189
  case "tool":
@@ -193,7 +203,6 @@ class GeminiManager:
193
  continue
194
  case "function_call":
195
  role = "model"
196
- print(message)
197
  formatted_history.append(
198
  eval(message.get("content", "")))
199
  continue
 
172
  if isinstance(message["content"], tuple):
173
  path = message["content"][0]
174
  try:
175
+ image_bytes = open(path, "rb").read()
176
+ parts = [
177
+ types.Part.from_bytes(
178
+ data=image_bytes,
179
+ mime_type="image/png",
180
+ ),
181
+ ]
182
  except Exception as e:
183
  logger.error(f"Error uploading file: {e}")
184
+ parts = [types.Part.from_text(
185
+ text="Error uploading file: "+str(e))]
186
+ formatted_history.append(
187
+ types.Content(
188
+ role=role,
189
+ parts=parts
190
+ ))
191
  continue
192
  else:
193
  parts = [types.Part.from_text(
194
  text=message.get("content", ""))]
195
  case "memories":
196
+ role = "model"
197
  parts = [types.Part.from_text(
198
  text="Relevant memories: "+message.get("content", ""))]
199
  case "tool":
 
203
  continue
204
  case "function_call":
205
  role = "model"
 
206
  formatted_history.append(
207
  eval(message.get("content", "")))
208
  continue