Merge branch 'main' of https://huggingface.co/spaces/Ekimetrics/climate-question-answering

climateqa/engine/talk_to_data/workflow.py

@@ -1,6 +1,6 @@
 import os
 
-from typing import Any, Callable, NotRequired, TypedDict
+from typing import Any, Callable, TypedDict, Optional
 import pandas as pd
 
 from plotly.graph_objects import Figure
@@ -35,9 +35,9 @@ class TableState(TypedDict):
     """
     table_name: str
     params: dict[str, Any]
-    sql_query: NotRequired[str]
-    dataframe: NotRequired[pd.DataFrame | None]
-    figure: NotRequired[Callable[..., Figure]]
+    sql_query: Optional[str]
+    dataframe: Optional[pd.DataFrame | None]
+    figure: Optional[Callable[..., Figure]]
     status: str
 
 class PlotState(TypedDict):
@@ -59,7 +59,7 @@ class State(TypedDict):
     user_input: str
     plots: list[str]
     plot_states: dict[str, PlotState]
-    error: NotRequired[str]
+    error: Optional[str]
     
 async def drias_workflow(user_input: str) -> State:
     """Performs the complete workflow of Talk To Drias : from user input to sql queries, dataframes and figures generated
@@ -187,8 +187,8 @@ async def find_param(state: State, param_name:str, table: str) -> dict[str, Any]
 
 class Location(TypedDict):
     location: str
-    latitude: NotRequired[str]
-    longitude: NotRequired[str]
+    latitude: Optional[str]
+    longitude: Optional[str]
 
 async def find_location(user_input: str, table: str) -> Location:
     print(f"---- Find location in table {table} ----")

requirements.txt

@@ -1,13 +1,13 @@
 gradio==5.0.2
 azure-storage-file-share==12.11.1
-azure-storage-blob
+azure-storage-blob==12.23.0
 python-dotenv==1.0.0
 langchain==0.2.1
 langchain_openai==0.1.7
 langgraph==0.2.70
 pinecone-client==4.1.0
 sentence-transformers==2.6.0
-huggingface-hub
+huggingface-hub==0.25.2
 pyalex==0.13
 networkx==3.2.1
 pyvis==0.3.2
@@ -20,4 +20,8 @@ msal==1.31
 matplotlib==3.9.2
 gradio-modal==0.0.4
 vanna==0.7.5
-geopy==2.4.1
+geopy==2.4.1
+duckdb==1.2.1
+openai==1.61.1
+pydantic==2.9.2
+pydantic-settings==2.2.1