thanhnt-cf commited on
Commit
71e2102
·
verified ·
1 Parent(s): 8c5ab45

hot fix issue after updating known data

Browse files
Files changed (1) hide show
  1. llm_api/native_api.py +2 -2
llm_api/native_api.py CHANGED
@@ -22,7 +22,7 @@ def extract_info(image_paths, provider, schema, known_data=None):
22
  return extractor.extract_info(image_paths, schema, known_data=known_data)
23
 
24
 
25
- def follow_structure(json_info, provider, schema):
26
  if provider == "anthropic":
27
  extractor = anthropic_extractor
28
  elif provider == "openai":
@@ -30,4 +30,4 @@ def follow_structure(json_info, provider, schema):
30
  else:
31
  raise ValueError("Invalid provider")
32
 
33
- return extractor.follow_structure(json_info, schema)
 
22
  return extractor.extract_info(image_paths, schema, known_data=known_data)
23
 
24
 
25
+ def follow_structure(json_info, provider, schema, known_data=None):
26
  if provider == "anthropic":
27
  extractor = anthropic_extractor
28
  elif provider == "openai":
 
30
  else:
31
  raise ValueError("Invalid provider")
32
 
33
+ return extractor.follow_structure(json_info, schema, known_data=known_data)