Initial commit

.gitignore

@@ -0,0 +1,4 @@
+venv/
+.env
+__pycache__/
+*.pyc

Dockerfile

@@ -0,0 +1,12 @@
+FROM python:3.10-slim       
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app/ ./app/
+
+ENV PYTHONPATH=/app
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

app/__init__.py

@@ -0,0 +1,5 @@
+"""
+Gmail Agent API using Gemini and Composio
+"""
+
+__version__ = "1.0.0"

app/agent.py

@@ -0,0 +1,52 @@
+# app/agent.py
+from llama_index.core import Settings
+from llama_index.llms.gemini import Gemini
+from llama_index.core.agent import FunctionCallingAgentWorker
+from composio_llamaindex import ComposioToolSet, App
+from llama_index.core.llms import ChatMessage
+
+class GmailAgent:
+    def __init__(self, google_api_key: str, composio_api_key: str):
+        # Set up Gemini LLM
+        Settings.llm = Gemini(
+            model="models/gemini-1.5-pro",
+            api_key=google_api_key
+        )
+
+        # Initialize Composio ToolSet
+        self.composio_toolset = ComposioToolSet(api_key=composio_api_key)
+        self.gmail_tools = self.composio_toolset.get_tools(apps=[App.GMAIL])
+
+        # Set up agent
+        prefix_messages = [
+            ChatMessage(
+                role="system",
+                content="""You are a Gmail Assistant that can:
+                - Read and analyze email content
+                - Draft professional replies
+                - Manage email threads
+                Use the available Gmail tools to perform these actions."""
+            )
+        ]
+
+        agent_worker = FunctionCallingAgentWorker.from_tools(
+            tools=self.gmail_tools,
+            prefix_messages=prefix_messages,
+            verbose=True
+        )
+        self.agent = agent_worker.as_agent()
+
+    def authenticate_gmail(self):
+        request = self.composio_toolset.initiate_connection(app=App.GMAIL)
+        return request.redirectUrl
+
+    async def process_email(self, thread_id: str, email_content: str, sender_email: str):
+        prompt = f"""Analyze this email and create an appropriate reply:
+        Thread ID: {thread_id}
+        Sender: {sender_email}
+        Content: {email_content}
+        
+        Please draft a professional response and use the appropriate Gmail tool to send it."""
+        
+        response = await self.agent.achat(prompt)
+        return str(response)

app/config.py

@@ -0,0 +1,10 @@
+# app/config.py
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Config:
+    GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+    COMPOSIO_API_KEY = os.getenv("COMPOSIO_API_KEY")
+    ALLOWED_ORIGINS = ["*"]  # Adjust for production

app/main.py

@@ -0,0 +1,69 @@
+# app/main.py
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from .config import Config
+from .agent import GmailAgent
+from app import __version__
+
+app = FastAPI(
+    title="Gmail Agent API",
+    description="API for Gmail management using Gemini and Composio",
+    version=__version__
+)
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=Config.ALLOWED_ORIGINS,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Initialize Gmail Agent
+gmail_agent = GmailAgent(
+    google_api_key=Config.GOOGLE_API_KEY,
+    composio_api_key=Config.COMPOSIO_API_KEY
+)
+
+class EmailRequest(BaseModel):
+    thread_id: str
+    email_content: str
+    sender_email: str
+
+@app.get("/")
+async def root():
+    """
+    Root endpoint that provides API information and health check
+    """
+    return {
+        "status": "online",
+        "version": __version__,
+        "name": "Gmail Agent API",
+        "description": "Gmail Agent API using Gemini and Composio"
+    }
+
+@app.get("/health")
+async def health_check():
+    return {"status": "healthy"}
+
+@app.get("/authenticate")
+async def get_auth_url():
+    try:
+        auth_url = gmail_agent.authenticate_gmail()
+        return {"auth_url": auth_url}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/process-email")
+async def process_email(request: EmailRequest):
+    try:
+        response = await gmail_agent.process_email(
+            thread_id=request.thread_id,
+            email_content=request.email_content,
+            sender_email=request.sender_email
+        )
+        return {"response": response}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

requirements.txt

@@ -0,0 +1,8 @@
+fastapi
+uvicorn
+python-dotenv
+composio-llamaindex
+llama-index
+llama-index-llms-gemini
+pydantic
+python-multipart