data-science-document-ai 1.42.0__py3-none-any.whl → 1.42.1__py3-none-any.whl

{data_science_document_ai-1.42.0.dist-info → data_science_document_ai-1.42.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-science-document-ai
-Version: 1.42.0
+Version: 1.42.1
 Summary: "Document AI repo for data science"
 Author: Naomi Nguyen
 Author-email: naomi.nguyen@forto.com

{data_science_document_ai-1.42.0.dist-info → data_science_document_ai-1.42.1.dist-info}/RECORD

@@ -3,7 +3,7 @@ src/constants_sandbox.py,sha256=Iu6HdjCoNSmOX0AwoL9qUQkhq_ZnIN5U9e-Q2UfNuGc,547
 src/docai.py,sha256=dHuR0ehVjUi1CnoNvdp_yxJtpU_HFXqAZ61ywdz7BEo,5655
 src/docai_processor_config.yaml,sha256=qOMmCIORpLQ_D-ytvejXxFvER0e0uGYuzPVdZBGv4Pc,2105
 src/excel_processing.py,sha256=jBL6h5T3fJ4uM_rFiV8c0yWAy8Tt3V3RFtBBqb8ztfo,2744
-src/io.py,sha256=OZh75Q9Fmh72kQz9Fj3L13CMTWJATqaloAb2eCWAu2U,4169
+src/io.py,sha256=tOJpMyI-mP1AaXKG4UFudH47MHWzjWBgVahFJUcjGfs,4749
 src/llm.py,sha256=OE4IEIqcM-hYK9U7e0x1rAfcqdpeo4iXPHBp64L5Qz0,8199
 src/log_setup.py,sha256=RhHnpXqcl-ii4EJzRt47CF2R-Q3YPF68tepg_Kg7tkw,2895
 src/pdf_processing.py,sha256=dxsYvNnONAjzS-T7K5aSo89rz7QcdW3ZDfeuFyeCeII,16294
@@ -52,6 +52,6 @@ src/prompts/prompt_library.py,sha256=jPxybNPPGH7mzonqtAOqmw5WcT-RtbGP0pvMqqP22hg
 src/setup.py,sha256=M-p5c8M9ejKcSZ9N86VtmtPc4TYLxe1_4_dxf6jpfVc,7262
 src/tms.py,sha256=UXbIo1QE--hIX6NZi5Qyp2R_CP338syrY9pCTPrfgnE,1741
 src/utils.py,sha256=nU69zR3TB7IZmCc19DD8H27Riek8GJAldmhJjCSwNEE,16090
-data_science_document_ai-1.42.0.dist-info/METADATA,sha256=IBhh-dmdAzKrTeE3ymx0ALVt7enKMmv0VADjUqtaMIE,2153
-data_science_document_ai-1.42.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-data_science_document_ai-1.42.0.dist-info/RECORD,,
+data_science_document_ai-1.42.1.dist-info/METADATA,sha256=nsGhuml2YNlNF7s7aRUJPpY8psKss8wiLcIavpVInjs,2153
+data_science_document_ai-1.42.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+data_science_document_ai-1.42.1.dist-info/RECORD,,

src/io.py

@@ -11,8 +11,6 @@ from pathlib import Path
 
 from google.cloud import bigquery, storage
 
-from src.constants import project_parameters
-
 
 def get_gcp_labels(**extra_labels):
     """Generate standardized GCP labels for cost tracking.
@@ -23,12 +21,34 @@ def get_gcp_labels(**extra_labels):
     Returns:
         dict: Labels dictionary with keys normalized (lowercase, hyphens, max 63 chars)
     """
+    project_name = os.getenv("PROJECT_NAME")
+
+    # If not set, detect once and cache it
+    if not project_name:
+        # Try pyproject.toml first
+        try:
+            import toml
+
+            pyproject_path = Path(__file__).parent.parent / "pyproject.toml"
+            if pyproject_path.exists():
+                config = toml.load(pyproject_path)
+                project_name = config.get("tool", {}).get("poetry", {}).get("name")
+        except Exception:
+            pass
+
+        # Fallback to unknown
+        if not project_name:
+            project_name = "unknown"
+
+        # Cache it
+        os.environ["PROJECT_NAME"] = project_name
+
     labels = {
-        "ds-project-name": project_parameters["project_name"],
+        "ds-project-name": project_name.lower(),
         "ds-env": os.getenv("CLUSTER", "local").lower(),
     }
 
-    # Add any extra labels passed in
+    # Add any extra labels
     labels.update({k.lower(): str(v).lower() for k, v in extra_labels.items()})
 
     return labels