update known data

llm_api/native_api.py

@@ -1,30 +1,33 @@
 import base64
+
 from dotenv import load_dotenv
-from . import openai_api as openai_info_extractor
+
 from . import anthropic_api as anthropic_extractor
+from . import openai_api as openai_info_extractor
+
 
 def encode_image(image_path):
-  with open(image_path, "rb") as image_file:
-    return base64.b64encode(image_file.read()).decode('utf-8')
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
 
 
-def extract_info(image_paths, provider, schema):
-    if provider == 'anthropic':
+def extract_info(image_paths, provider, schema, known_data=None):
+    if provider == "anthropic":
         extractor = anthropic_extractor
-    elif provider == 'openai':
+    elif provider == "openai":
         extractor = openai_info_extractor
     else:
-        raise ValueError('Invalid provider')
-    
-    return extractor.extract_info(image_paths, schema)
+        raise ValueError("Invalid provider")
+
+    return extractor.extract_info(image_paths, schema, known_data=known_data)
 
 
 def follow_structure(json_info, provider, schema):
-    if provider == 'anthropic':
+    if provider == "anthropic":
         extractor = anthropic_extractor
-    elif provider == 'openai':
+    elif provider == "openai":
         extractor = openai_info_extractor
     else:
-        raise ValueError('Invalid provider')
-    
-    return extractor.follow_structure(json_info, schema)
+        raise ValueError("Invalid provider")
+
+    return extractor.follow_structure(json_info, schema)