Create settings_ai.py

settings_ai.py

@@ -0,0 +1,71 @@
+from google.ai.generativelanguage_v1beta.types import content
+from typing import List, Optional
+from pydantic import BaseModel, ValidationError
+from dotenv import load_dotenv
+load_dotenv()
+import os 
+
+ENDPOINT_AZURE = os.getenv("ENDPOINT_AZURE")
+API_AZURE = os.getenv("API_AZURE")
+API_KEY_GEMINI = os.getenv("API_KEY_GEMINI")
+NO_ARTICLE = "NESSUN_ARTICOLO_COMPATIBILE"
+SYSTEM_INSTRUCTION = (
+    "Sei un esperto lettore di documenti. "
+    "Estrai il tutto il formato JSON come da schema fornito! "
+    "Mi raccomando compila tutti i campi! "
+    "Estra SOLO gli Articoli il cui codice inizia per 'AVE' o 'AV' o '3V' o '44'. Attenzione a quelli che iniziano per 'GEW 44..' sono da ESCLUDERE! "
+    "Nel campo 'TotaleNonIvato' inserisci il TOTALE MERCE per riga senza Iva: (Quantità * Prezzo Unitario) - Sconto. "
+    "Potrebbe non esserci lo 'Sconto' oppure potrebbe esserci direttamente il TOTALE MERCE per riga. "
+    "La 'Data' documento mettila in formato YYYY-MM-DD. "
+    "Attenzione, a volte ci sono articoli con DESCRIZIONE AVE ma con codice NON CORRISPONDENTE, devi escluderli! "
+    "Attenzione, a volte gli articoli AVE sono dentro la parentesi per esempio 'A789532 (AVE001)', qui l'articolo risulta COMPATIBILE perché ha AVE! " 
+    "Attenzione, a volte i TOTALI hanno la formattazione con il punto (.) che rappresenta il SEPARATORE DELLE MIGLIAIA! Quindi 1.234,50 è 1234,50! "
+    f"Se non sono presenti articoli compatibili scrivi '{NO_ARTICLE}' nel codice articolo e metti 0 nel 'TotaleNonIvato'"
+)
+
+USER_MESSAGE = (
+    "Estrai i dati in formato JSON. "
+)
+
+class Articolo(BaseModel):
+    CodiceArticolo: str
+    DescrizioneArticolo: str
+    TotaleNonIvato: float
+    Verificato: Optional[int] = None
+
+class Documento(BaseModel):
+    TipoDocumento: str
+    NumeroDocumento: str
+    Data: str
+    TotaleImponibile: float
+    Articoli: List[Articolo]
+
+GENERATION_CONFIG = {
+    "temperature": 0.01,
+    "top_p": 0.1, 
+    "top_k": 5,   
+    "max_output_tokens": 500000,
+    "response_schema": content.Schema(
+        type=content.Type.OBJECT,
+        required=["TipoDocumento", "NumeroDocumento", "Data", "TotaleImponibile"],
+        properties={
+            "TipoDocumento": content.Schema(type=content.Type.STRING, enum=["FATTURA", "DDT"]),
+            "NumeroDocumento": content.Schema(type=content.Type.STRING),
+            "Data": content.Schema(type=content.Type.STRING),
+            "TotaleImponibile": content.Schema(type=content.Type.NUMBER),
+            "Articoli": content.Schema(
+                type=content.Type.ARRAY,
+                items=content.Schema(
+                    type=content.Type.OBJECT,
+                    required=["CodiceArticolo", "DescrizioneArticolo", "TotaleNonIvato"],
+                    properties={
+                        "CodiceArticolo": content.Schema(type=content.Type.STRING),
+                        "DescrizioneArticolo": content.Schema(type=content.Type.STRING),
+                        "TotaleNonIvato": content.Schema(type=content.Type.NUMBER),
+                    },
+                ),
+            )
+        },
+    ),
+    "response_mime_type": "application/json",
+}