Update weaviate_utils.py

weaviate_utils.py

@@ -146,7 +146,12 @@ def get_class_schema(client, class_name):
 
 def retrieve_relevant_table(client, selected_class, question_embedding):
     # Query Weaviate to get the most relevant table
-    results = client.query.get(selected_class, ["embeddedTable"]).with_near_text(question_embedding).do()
-    # Extract the table from the results
-    table = results.get('data', {}).get('Get', {}).get('Things', [])[0].get('embeddedTable')
+    results = client.query.get(selected_class, ["content"]).with_near_text(question_embedding).do()
+    
+    # Extract the table content from the results
+    table_content = results.get('data', {}).get('Get', {}).get('Things', [])[0].get('content')
+    
+    # Convert the table content to a DataFrame
+    table = pd.read_csv(StringIO(table_content))
+    
     return table