anmolsahai commited on
Commit
eeae563
1 Parent(s): 855b4b0
__pycache__/langchain_pipeline.cpython-310.pyc CHANGED
Binary files a/__pycache__/langchain_pipeline.cpython-310.pyc and b/__pycache__/langchain_pipeline.cpython-310.pyc differ
 
langchain_pipeline.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- from pdfminer import high_level
3
  from langchain_astradb import AstraDBVectorStore
4
  from langchain_core.prompts import PromptTemplate
5
  from langchain_openai import OpenAIEmbeddings
@@ -25,7 +25,7 @@ def model_names():
25
  return models.keys()
26
 
27
  def pipeline(file, model_name, balance_type, apsn_transactions, max_fees_per_day, min_overdrawn_fee, min_transaction_overdraft):
28
- disclosure_text = high_level.extract_text(file)
29
  astra = AstraDBVectorStore(
30
  api_endpoint=ASTRA_DB_API_ENDPOINT,
31
  token=ASTRA_DB_APPLICATION_TOKEN,
 
1
  import os
2
+ from pdfminer.high_level import extract_text
3
  from langchain_astradb import AstraDBVectorStore
4
  from langchain_core.prompts import PromptTemplate
5
  from langchain_openai import OpenAIEmbeddings
 
25
  return models.keys()
26
 
27
  def pipeline(file, model_name, balance_type, apsn_transactions, max_fees_per_day, min_overdrawn_fee, min_transaction_overdraft):
28
+ disclosure_text = extract_text(file)
29
  astra = AstraDBVectorStore(
30
  api_endpoint=ASTRA_DB_API_ENDPOINT,
31
  token=ASTRA_DB_APPLICATION_TOKEN,