File size: 403 Bytes
b2b9299
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from weather_node import fetch_weather
from rag_node import rag_answer

def decide_and_execute(query: str) -> str:
    """Decide if query is about weather or PDF RAG."""
    keywords = ["weather", "temperature", "forecast", "climate"]
    if any(word in query.lower() for word in keywords):
        return fetch_weather(query)   # query is passed directly now
    else:
        return rag_answer(query)