from typing import TypedDict, List
from langchain_core.messages import AnyMessage

class JARVISState(TypedDict):
    task_id: str
    question: str
    tools_needed: List[str]
    web_results: List[str]
    file_results: str
    image_results: str
    calculation_results: str
    document_results: str
    messages: List[AnyMessage]
    answer: str