feat: pandasai openai llm
Browse files- postgre_mcp_server.py +4 -0
- requirements.txt +0 -0
postgre_mcp_server.py
CHANGED
|
@@ -627,7 +627,11 @@ async def visualize_results(json_data: dict, vis_prompt: str) -> str:
|
|
| 627 |
"""
|
| 628 |
try:
|
| 629 |
# Debug prints to see what's being received
|
|
|
|
| 630 |
|
|
|
|
|
|
|
|
|
|
| 631 |
# Convert JSON to DataFrame
|
| 632 |
df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
|
| 633 |
|
|
|
|
| 627 |
"""
|
| 628 |
try:
|
| 629 |
# Debug prints to see what's being received
|
| 630 |
+
from pandasai_openai import OpenAI
|
| 631 |
|
| 632 |
+
pllm = OpenAI(api_token=os.environ["OPENAI_API_KEY"])
|
| 633 |
+
pai.config.set({"llm": pllm})
|
| 634 |
+
|
| 635 |
# Convert JSON to DataFrame
|
| 636 |
df = pd.DataFrame(json_data["data"], columns=json_data["columns"])
|
| 637 |
|
requirements.txt
CHANGED
|
Binary files a/requirements.txt and b/requirements.txt differ
|
|
|