cevheri commited on
Commit
685f01f
·
1 Parent(s): a2d0bfa

feat: pandasai openai llm

Browse files
Files changed (2) hide show
  1. postgre_mcp_server.py +4 -0
  2. 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