helloparthshah commited on
Commit
e162f30
·
1 Parent(s): baea1e6

Fixing a bug with the manager

Browse files
Files changed (1) hide show
  1. src/manager/manager.py +2 -1
src/manager/manager.py CHANGED
@@ -176,7 +176,8 @@ class GeminiManager:
176
  formatted_history = []
177
  for message in messages:
178
  # Skip thinking messages (messages with metadata)
179
- if not (message.get("role") == "assistant" and "metadata" in message):
 
180
  role = "model"
181
  match message.get("role"):
182
  case "user":
 
176
  formatted_history = []
177
  for message in messages:
178
  # Skip thinking messages (messages with metadata)
179
+ if not ((message.get("role") == "assistant" and "metadata" in message
180
+ and message["metadata"] is not None)):
181
  role = "model"
182
  match message.get("role"):
183
  case "user":