add google big query library
Browse files- .gitignore +8 -0
- app.py +2 -2
.gitignore
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# add .venv, python cache files to .gitignore
|
| 2 |
+
.venv/
|
| 3 |
+
__pycache__/
|
| 4 |
+
|
| 5 |
+
# add compiled python files to .gitignore
|
| 6 |
+
*.pyc
|
| 7 |
+
crisis_prediction.db
|
| 8 |
+
setup_duckdb_crisis_prediction.py
|
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import gradio as gr
|
|
| 3 |
from datasets import load_dataset
|
| 4 |
from typing import Union, Dict, List, Optional
|
| 5 |
import pandas as pd
|
| 6 |
-
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
| 7 |
from opentelemetry import trace
|
| 8 |
from opentelemetry.sdk.trace import TracerProvider
|
| 9 |
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
|
@@ -92,7 +92,7 @@ model = HfApiModel()
|
|
| 92 |
agent = CodeAgent(
|
| 93 |
tools=[],
|
| 94 |
model=model,
|
| 95 |
-
additional_authorized_imports=["gradio","pandas","datasets","numpy","Bio","rdflib","networkx","requests","execute_sql","duckdb","pyarrow"],
|
| 96 |
add_base_tools=True
|
| 97 |
)
|
| 98 |
|
|
|
|
| 3 |
from datasets import load_dataset
|
| 4 |
from typing import Union, Dict, List, Optional
|
| 5 |
import pandas as pd
|
| 6 |
+
from smolagents import CodeAgent, HfApiModel, tool, GradioUI
|
| 7 |
from opentelemetry import trace
|
| 8 |
from opentelemetry.sdk.trace import TracerProvider
|
| 9 |
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
|
|
|
| 92 |
agent = CodeAgent(
|
| 93 |
tools=[],
|
| 94 |
model=model,
|
| 95 |
+
additional_authorized_imports=["pyvis","aiohttp","os","sys","gradio","pandas","datasets","numpy","Bio","rdflib","networkx","requests","execute_sql","duckdb","pyarrow","google-cloud-bigquery"],
|
| 96 |
add_base_tools=True
|
| 97 |
)
|
| 98 |
|