Upload 2 files

functions/sqlite_functions.py

@@ -11,11 +11,12 @@ class SQLiteQuery:
 
     @component.output_types(results=List[str], queries=List[str])
     def run(self, queries: List[str]):
+        print("ATTEMPTING TO RUN QUERY")
         results = []
         for query in queries:
           result = pd.read_sql(query, self.connection)
           results.append(f"{result}")
-        self.connection.close()  
+        "self.connection.close()"
         return {"results": results, "queries": queries}
     
 

pipelines/pipelines.py

@@ -71,10 +71,11 @@ conditional_sql_pipeline.connect("router.sql", "sql_querier.queries")
 conditional_sql_pipeline.connect("router.go_to_fallback", "fallback_prompt.question")
 conditional_sql_pipeline.connect("fallback_prompt", "fallback_llm")
 
-def rag_pipeline_func(question: str, columns: str):
-   result = conditional_sql_pipeline.run({"prompt": {"question": question,
+def rag_pipeline_func(queries: str, columns: str):
+   print("RAG PIPELINE FUNCTION")
+   result = conditional_sql_pipeline.run({"prompt": {"question": queries,
                                                   "columns": columns},
-                                       "router": {"question": question},
+                                       "router": {"question": queries},
                                        "fallback_prompt": {"columns": columns}})
    
    if 'sql_querier' in result: