cccc

my_tools.py

@@ -1,6 +1,7 @@
 import os, math, wikipedia, asyncio
 import google.generativeai as genai
 from typing import List, Generator
+from llama_index.core.llms import LLMMetadata
 
 from llama_index.core.llms import (
     LLM,
@@ -25,13 +26,13 @@ class GeminiLLM(LLM):
 
     # ---------- metadata ----------
     @property
-    def metadata(self):
-        return {
-            "context_window": 32768,
-            "num_output": 2048,
-            "is_chat_model": True,
-            "model_name": self._model_name,
-        }
+    def metadata(self) -> LLMMetadata:
+        return LLMMetadata(
+            context_window=32768,
+            num_output=2048,
+            is_chat_model=True,
+            model_name=self._model_name
+        )
 
     # ---------- chat ----------
     def chat(self, messages: List[ChatMessage], **kwargs) -> ChatResponse: