helloparthshah commited on
Commit
b746682
·
1 Parent(s): 60ee681

disabling tool invocation stops sending it the tools

Browse files
Files changed (1) hide show
  1. src/manager/manager.py +4 -1
src/manager/manager.py CHANGED
@@ -91,7 +91,10 @@ class GeminiManager:
91
  max_tries=3,
92
  jitter=None)
93
  def generate_response(self, messages):
94
- tools = self.toolsLoader.getTools()
 
 
 
95
  return self.client.models.generate_content_stream(
96
  model=self.model_name,
97
  contents=messages,
 
91
  max_tries=3,
92
  jitter=None)
93
  def generate_response(self, messages):
94
+ if self.check_mode(Mode.ENABLE_TOOL_INVOCATION):
95
+ tools = self.toolsLoader.getTools()
96
+ else:
97
+ tools = []
98
  return self.client.models.generate_content_stream(
99
  model=self.model_name,
100
  contents=messages,