"""Weather Agent - LangGraph"""
import os
from typing import TypedDict
from langgraph.graph import StateGraph, END
from langchain_openai import ChatOpenAI
import requests

OPENAI_KEY = os.getenv("OPENAI_API_KEY")
WEATHER_KEY = os.getenv("WEATHER_API_KEY")

class WeatherState(TypedDict):
    city: str
    weather: dict
    response: str

class WeatherAgent:
    def __init__(self):
        self.llm = ChatOpenAI(
            model="gpt-4",
            api_key=OPENAI_KEY
        )
    
    def fetch(self, state: WeatherState):
        city = state["city"]
        url = f"https://api.openweathermap.org"
        url += f"/data/2.5/weather?q={city}"
        url += f"&appid={WEATHER_KEY}&units=metric"
        
        try:
            r = requests.get(url)
            state["weather"] = r.json()
        except Exception as e:
            state["weather"] = {"error": str(e)}
        return state
    
    def respond(self, state: WeatherState):
        data = state["weather"]
        if "error" in data:
            state["response"] = "Error!"
            return state
        
        prompt = f"""Weather for {state['city']}:
Temp: {data['main']['temp']}°C
Condition: {data['weather'][0]['description']}
Humidity: {data['main']['humidity']}%"""
        
        resp = self.llm.invoke(prompt)
        state["response"] = resp.content
        return state
    
    def build(self):
        graph = StateGraph(WeatherState)
        graph.add_node("fetch", self.fetch)
        graph.add_node("respond", self.respond)
        graph.set_entry_point("fetch")
        graph.add_edge("fetch", "respond")
        graph.add_edge("respond", END)
        return graph.compile()

def get_weather_langgraph(city: str):
    agent = WeatherAgent()
    graph = agent.build()
    result = graph.invoke({
        "city": city,
        "weather": {},
        "response": ""
    })
    return result["response"]

agent_name: My Agent
description: A weather checking agent
framework: langgraph
template: default
version: 1.0.0
template_source:
  repo_url: https://github.com/runagent-dev/runagent.git
  path: templates/langgraph
  author: sawradip
agent_architecture:
  entrypoints:
    - file: agent.py
      module: get_weather_langgraph
      tag: weather_info

from runagent import RunAgentClient

# Create a client pointing to your agent
weather_agent = RunAgentClient(
    agent_id="XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX",
    entrypoint_tag="weather_info"
)

# Call your agent
agent_result = weather_agent.run(
    city="New York"
)

print(agent_result)