Update ocr_llm_utils.py

ocr_llm_utils.py

@@ -28,7 +28,7 @@ def run_ocr_with_gcv(image_path):
     response = client_vision.document_text_detection(image=image)
     return response.full_text_annotation.text
 
-def extract_table_from_text(text):
+def extract_table_from_text(text,max_tokens=4096):
     prompt = f"""
     Extract a structured table of items from the invoice text below. 
     - First findout what are the table column names
@@ -50,7 +50,7 @@ def extract_table_from_text(text):
         {"role": "user", "content": prompt}
     ],
         temperature=1,
-        max_completion_tokens=4096,
+        max_completion_tokens=max_tokens,
         top_p=1,
     )
     return response.choices[0].message.content