bugfix

langchain_pipeline.py

@@ -22,15 +22,11 @@ models = {
     "gemini-pro": ChatGoogleGenerativeAI(model="gemini-pro")
 }
 
-
 def model_names():
     return models.keys()
 
-
-def pipeline(bytes, model_name):
-    disclosure_text = high_level.extract_text(bytes)
-#     disclosure_text = doc[0].page_content
-#
+def pipeline(file, model_name, balance_type, apsn_transactions, max_fees_per_day, min_overdrawn_fee, min_transaction_overdraft):
+    disclosure_text = high_level.extract_text(file)
     astra = AstraDBVectorStore(
         api_endpoint=ASTRA_DB_API_ENDPOINT,
         token=ASTRA_DB_APPLICATION_TOKEN,
@@ -42,6 +38,7 @@ def pipeline(bytes, model_name):
 
     prompt = PromptTemplate.from_template(
         """
+
     law context:
 
      20-cv-12061 08-23-2021