[Fix] Fix tools calls issue with openai models (#73)

Fix tools calls issue with openai models

Co-authored-by: Vipul Mittal <[email protected]>

Author: psriramsnc

pyproject.toml

@@ -71,6 +71,7 @@ ui = [
   "streamlit-autorefresh>=1.0,<2.0",
   "streamlit-flow-component>=1.6,<2.0",
   "watchdog>=6.0,<7.0",
+  "plotly>=6.5.0"
 ]
 dev = [
   "pytest>=8.4,<9.0",

sygra/core/models/lite_llm/azure_openai_model.py

@@ -74,7 +74,7 @@ async def _generate_response(
         elif should_route_to_image(self.model_config):
             return await self._generate_image(input, model_params)
         else:
-            return await self._generate_text(input, model_params, **kwargs)
+            return await self._generate_text(input, model_params)
 
     async def _generate_speech(
         self, input: ChatPromptValue, model_params: ModelParams, **kwargs: Any

sygra/core/models/lite_llm/openai_model.py

@@ -76,7 +76,7 @@ async def _generate_response(
         elif should_route_to_image(self.model_config):
             return await self._generate_image(input, model_params)
         else:
-            return await self._generate_text(input, model_params, **kwargs)
+            return await self._generate_text(input, model_params)
 
     async def _generate_speech(
         self, input: ChatPromptValue, model_params: ModelParams, **kwargs: Any