| from google import genai |
| from google.genai import types |
|
|
| class Google_Gemini_LLM: |
| def __init__(self, api_key, model="gemini-1.5-flash"): |
| self.api_key = api_key |
| self.model = model |
| self.client = genai.Client(api_key=api_key) |
| self.LLM_tools = [] |
|
|
| def generate_content(self, contents): |
| tools = types.Tool(function_declarations=self.LLM_tools) |
| config = types.GenerateContentConfig(tools=[tools]) |
| response = self.client.models.generate_content( |
| model=self.model, contents=contents, config=config |
| ) |
| return response.candidates[0].content.parts[0] |
|
|
| def append_function_tools(self, tools): |
| for tool in tools: |
| self.LLM_tools.append(tool) |
|
|
