use premium (#10)

adrianlyjak · web-flow · commit 87c2da862d77 · 2025-08-07T11:34:27.000-04:00
diff --git a/src/{{ project_name_snake }}/config.py.jinja b/src/{{ project_name_snake }}/config.py.jinja
@@ -28,7 +28,7 @@ extracted_data_collection = "{{ project_name }}"
 def get_extract_agent() -> ExtractionAgent:
     extract_api = LlamaExtract(api_key=api_key, base_url=base_url)
     config = ExtractConfig(
-        extraction_mode=ExtractMode.BALANCED,
+        extraction_mode=ExtractMode.PREMIUM,
         system_prompt=None,
         # advanced
         use_reasoning=False,
diff --git a/test-proj/src/test_proj/config.py b/test-proj/src/test_proj/config.py
@@ -28,7 +28,7 @@
 def get_extract_agent() -> ExtractionAgent:
     extract_api = LlamaExtract(api_key=api_key, base_url=base_url)
     config = ExtractConfig(
-        extraction_mode=ExtractMode.BALANCED,
+        extraction_mode=ExtractMode.PREMIUM,
         system_prompt=None,
         # advanced
         use_reasoning=False,
