gemini generate content with new skd, new model

Author: martinibach

community/gemini_generate/.env.Example community/gemini/.env.Examplecommunity/gemini_generate/.env.Example renamed to community/gemini/.env.Example

@@ -1,7 +1,7 @@
 [tool.poetry]
-name = "gemini_generate"
+name = "gemini"
 version = "0.0.1"
-description = "A simple example to generate content using Google Gemini"
+description = "A simple example to integrate Google Gemini into a Restack project"
 authors = [
     "Restack Team <[email protected]>",
 ]
@@ -11,7 +11,7 @@ packages = [{include = "src"}]
 [tool.poetry.dependencies]
 python = ">=3.10,<4.0"
 restack-ai = "^0.0.52"
-google-generativeai = "0.8.3"
+google-genai = "0.5.0"
 watchfiles = "^1.0.0"
 
 [build-system]

community/gemini_generate/.gitignore community/gemini/.gitignorecommunity/gemini_generate/.gitignore renamed to community/gemini/.gitignore

@@ -9,15 +9,15 @@ class FunctionInputParams:
     user_content: str
 
 @function.defn()
-async def gemini_generate_opposite(input: FunctionInputParams) -> str:
+async def gemini_generate_content(input: FunctionInputParams) -> str:
     try:
-        log.info("gemini_generate_opposite function started", input=input)
+        log.info("gemini_generate_content function started", input=input)
         genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
-        model = genai.GenerativeModel("gemini-1.5-flash")
+        model = genai.GenerativeModel("gemini-2.0-flash-exp")
 
         response = model.generate_content(input.user_content)
-        log.info("gemini_generate_opposite function completed", response=response.text)
+        log.info("gemini_generate_content function completed", response=response.text)
         return response.text
     except Exception as e:
-        log.error("gemini_generate_opposite function failed", error=e)
+        log.error("gemini_generate_content function failed", error=e)
         raise e