readgrid 0.1.0__tar.gz → 0.1.1__tar.gz

{readgrid-0.1.0 → readgrid-0.1.1}/LICENSE

@@ -1,5 +1,7 @@
 MIT License
 
+Copyright (c) 2025 David K. Jeremiah
+
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
@@ -17,5 +19,3 @@ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
-
-© 2025 David

{readgrid-0.1.0/readgrid.egg-info → readgrid-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: readgrid
-Version: 0.1.0
+Version: 0.1.1
 Summary: A document layout pipeline for detecting tables, images, and structured extraction.
 Home-page: https://github.com/davidkjeremiah/readgrid
 Author: David Jeremiah
@@ -110,4 +110,4 @@ stage_3(api_key="YOUR_API_KEY")
 ## 📄 License
 
 MIT License.
-See [LICENSE](LICENSE) for details.
+See LICENSE for details.

{readgrid-0.1.0 → readgrid-0.1.1}/README.md

@@ -80,4 +80,4 @@ stage_3(api_key="YOUR_API_KEY")
 ## 📄 License
 
 MIT License.
-See [LICENSE](LICENSE) for details.
+See LICENSE for details.

{readgrid-0.1.0 → readgrid-0.1.1}/readgrid/pipeline.py

@@ -679,7 +679,8 @@ def stage_3(
     api_key: Optional[str] = None, 
     custom_system_prompt: Optional[str] = None,
     output_fields: Optional[List[str]] = None,
-    exclude_fields: Optional[List[str]] = None
+    exclude_fields: Optional[List[str]] = None,
+    model_name: Optional[str] = None,
 ):
     """
     Processes annotated images through LLM with customizable JSON output.
@@ -712,7 +713,11 @@ def stage_3(
 
     print(f"ℹ️  Final JSON will include: {fields_to_include}")
 
-    # --- 2. Configure Gemini API ---
+    # Determine model
+    chosen_model = model_name or "gemini-1.5-flash"
+    print(f"ℹ️  Using model: {chosen_model}")
+
+    # --- 2. Configure Model API ---
     if not api_key:
         try:
             api_key = getpass("🔑 Please enter your Model's API Key: ")
@@ -788,7 +793,7 @@ def stage_3(
 
     # --- 4. Initialize Model and Load Data ---
     model = genai.GenerativeModel(
-        model_name='gemini-1.5-flash',
+        model_name=chosen_model,
         system_instruction=system_prompt
     )
     

{readgrid-0.1.0 → readgrid-0.1.1/readgrid.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: readgrid
-Version: 0.1.0
+Version: 0.1.1
 Summary: A document layout pipeline for detecting tables, images, and structured extraction.
 Home-page: https://github.com/davidkjeremiah/readgrid
 Author: David Jeremiah
@@ -110,4 +110,4 @@ stage_3(api_key="YOUR_API_KEY")
 ## 📄 License
 
 MIT License.
-See [LICENSE](LICENSE) for details.
+See LICENSE for details.

{readgrid-0.1.0 → readgrid-0.1.1}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="readgrid",
-    version="0.1.0",
+    version="0.1.1",
     author="David Jeremiah",
     author_email="flasconnect@gmail.com",
     description="A document layout pipeline for detecting tables, images, and structured extraction.",