Nancy1906 commited on
Commit
353ccd4
·
verified ·
1 Parent(s): b12d717
Files changed (1) hide show
  1. my_tools.py +2 -1
my_tools.py CHANGED
@@ -21,7 +21,8 @@ from llama_index.core.callbacks.llama_debug import LlamaDebugHandler
21
  # -------------------------------------------------------------------
22
  # 1) GeminiLLM personalizado
23
  # -------------------------------------------------------------------
24
- ChatMessage.message = property(lambda self: self.content)
 
25
  class GeminiLLM(LLM):
26
  model_name: str = Field(default="models/gemini-1.5-flash-latest")
27
  temperature: float = Field(default=0.0)
 
21
  # -------------------------------------------------------------------
22
  # 1) GeminiLLM personalizado
23
  # -------------------------------------------------------------------
24
+ # Monkey-patch para exponer .message al objeto completo, tal como espera LlamaIndex
25
+ ChatMessage.message = property(lambda self: self)
26
  class GeminiLLM(LLM):
27
  model_name: str = Field(default="models/gemini-1.5-flash-latest")
28
  temperature: float = Field(default=0.0)