Merge pull request #128 from restackio/gemini-fix

Author: martinibach

gemini_generate/Dockerfile

@@ -0,0 +1,22 @@
+FROM python:3.12-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y
+
+RUN pip install poetry
+
+COPY pyproject.toml ./
+
+COPY . .
+
+# Configure poetry to not create virtual environment
+RUN poetry config virtualenvs.create false
+
+# Install dependencies
+RUN poetry install --no-interaction --no-ansi
+
+# Expose port 80
+EXPOSE 80
+
+CMD poetry run python -m src.services

gemini_generate/pyproject.toml

@@ -10,13 +10,15 @@ packages = [{include = "src"}]
 
 [tool.poetry.dependencies]
 python = ">=3.10,<4.0"
-restack-ai = "^0.0.48"
+restack-ai = "^0.0.52"
 google-generativeai = "0.8.3"
+watchfiles = "^1.0.0"
 
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
+dev = "src.services:watch_services"
 services = "src.services:run_services"
 schedule = "schedule_workflow:run_schedule_workflow"

gemini_generate/src/client.py

@@ -1,3 +1,20 @@
+import os
 from restack_ai import Restack
+from restack_ai.restack import CloudConnectionOptions
+from dotenv import load_dotenv  
+# Load environment variables from a .env file
+load_dotenv()
 
-client = Restack()
+
+engine_id = os.getenv("RESTACK_ENGINE_ID")
+address = os.getenv("RESTACK_ENGINE_ADDRESS")
+api_key = os.getenv("RESTACK_ENGINE_API_KEY")
+api_address = os.getenv("RESTACK_ENGINE_API_ADDRESS")
+
+connection_options = CloudConnectionOptions(
+    engine_id=engine_id,
+    address=address,
+    api_key=api_key,
+    api_address=api_address
+)
+client = Restack(connection_options)

gemini_generate/src/services.py

@@ -1,4 +1,8 @@
 import asyncio
+from watchfiles import run_process
+import webbrowser
+import os
+
 from src.client import client
 from src.functions.function import gemini_generate_opposite
 from src.workflows.gemini_generate_content import GeminiGenerateOppositeWorkflow
@@ -9,7 +13,16 @@ async def main():
     )
 
 def run_services():
-    asyncio.run(main())
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("Service interrupted by user. Exiting gracefully.")
+
+def watch_services():
+    watch_path = os.getcwd()
+    print(f"Watching {watch_path} and its subdirectories for changes...")
+    webbrowser.open("http://localhost:5233")
+    run_process(watch_path, recursive=True, target=run_services)
 
 if __name__ == "__main__":
-    run_services()
+       run_services()