Spaces:
Runtime error
Runtime error
vvvv
Browse files- 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 |
-
|
|
|
|
| 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)
|