Nancy1906 commited on
Commit
946bca1
·
verified ·
1 Parent(s): ce8324a
Files changed (1) hide show
  1. my_tools.py +4 -0
my_tools.py CHANGED
@@ -111,6 +111,10 @@ class GeminiLLM(LLM):
111
  async def astream_complete(self, prompt, formatted=False, **kwargs):
112
  return await asyncio.to_thread(self.stream_complete, prompt, formatted=formatted, **kwargs)
113
 
 
 
 
 
114
 
115
  # -------------------------------------------------------------------
116
  # 2) Herramientas
 
111
  async def astream_complete(self, prompt, formatted=False, **kwargs):
112
  return await asyncio.to_thread(self.stream_complete, prompt, formatted=formatted, **kwargs)
113
 
114
+ # <<< METODO ADICIONAL PARA CUMPLIR con LLM.abstractmethod >>>
115
+ def astream_chat(self, messages: list[ChatMessage], **kwargs):
116
+ # reutilizamos stream_chat existente
117
+ return self.stream_chat(messages, **kwargs)
118
 
119
  # -------------------------------------------------------------------
120
  # 2) Herramientas