initial commit

app.py

@@ -36,8 +36,19 @@ def call_llm_api(input_text):
     return response.json()  # Adjust as needed based on your API response format
 	
 
+from langchain.llms import Runnable
+
+class APIRunnable(Runnable):
+    def __init__(self, api_func):
+        self.api_func = api_func
+
+    def run(self, input_text):
+        return self.api_func(input_text)
+
+api_runnable = APIRunnable(api_func=call_llm_api)
+
 rag_pipeline = RetrievalQA.from_chain_type(
-    llm=call_llm_api, chain_type='stuff',
+    llm=api_runnable, chain_type='stuff',
     retriever=vectorstore.as_retriever()
 )