update

.gitignore

@@ -0,0 +1 @@
+langchain_pipeline copy.py

app.py

@@ -46,7 +46,7 @@ def redline_changes(original_path, revised_path, output_path):
 st.title("Canarie AI Prototype")
 st.subheader("Finding the canarie in the coal mine")
 
-model_name = st.selectbox("Model", model_names())
+model_name = st.selectbox("Model", ["gemini-1.5-pro-001", "other-model-name"])
 
 balance_type = st.selectbox("Do you charge on available balance or ledger balance?", ["available balance", "ledger balance"])
 

langchain_pipeline.py

@@ -11,19 +11,17 @@ from vertexai.generative_models import GenerativeModel, Part, FinishReason
 import vertexai.preview.generative_models as generative_models
 
 def generate():
-  vertexai.init(project="akroda", location="us-central1")
-  model = GenerativeModel(
-    "gemini-1.5-pro-001",
-  )
-  responses = model.generate_content(
-      [document1, document2, document3, document4, document5, document6, document7, document8, document9, text1],
-      generation_config=generation_config,
-      safety_settings=safety_settings,
-      stream=True,
-  )
+    vertexai.init(project="akroda", location="us-central1")
+    model = GenerativeModel("gemini-1.5-pro-001")
+    responses = model.generate_content(
+        [document1, document2, document3, document4, document5, document6, document7, document8, document9, text1],
+        generation_config=generation_config,
+        safety_settings=safety_settings,
+        stream=True,
+    )
 
-  for response in responses:
-    print(response.text, end="")
+    for response in responses:
+        print(response.text, end="")
 
 document1 = Part.from_data(
     mime_type="application/pdf",
@@ -59,17 +57,15 @@ Please output in the following format:
 {entire updated disclosure text with changes bolded}
 ------
 {reasons for each change listed and cases cited}"""
-
-    )
-    val = prompt.format(
-        context=legal_cases_context,
-        disclosure=disclosure_text,
-        balance_type=balance_type,
-        apsn_transactions=apsn_transactions,
-        max_fees_per_day=max_fees_per_day,
-        min_overdrawn_fee=min_overdrawn_fee,
-        min_transaction_overdraft=min_transaction_overdraft,
-    )
+val = prompt.format(
+    context=legal_cases_context,
+    disclosure=disclosure_text,
+    balance_type=balance_type,
+    apsn_transactions=apsn_transactions,
+    max_fees_per_day=max_fees_per_day,
+    min_overdrawn_fee=min_overdrawn_fee,
+    min_transaction_overdraft=min_transaction_overdraft,
+)
 
 generation_config = {
     "max_output_tokens": 8192,
@@ -87,3 +83,4 @@ safety_settings = {
 generate()
 
 return chat_response.content
+