import os import json from dotenv import load_dotenv try: load_dotenv('.env') except: pass PACKAGE = 'invoice_extractor' PROJECT_DIR = os.getcwd() PACKAGE_PATH = os.path.join(PROJECT_DIR, PACKAGE) PROMPTS_DIR = os.path.join(PACKAGE_PATH, 'prompts') DATA_DIR = os.path.join(PACKAGE_PATH, 'data') CREDENTIALS = { 'azure' : { 'plain-text' : { 'endpoint' : os.environ.get('AZURE_PLAIN_TEXT_ENDPOINT', ''), 'key' : os.environ.get('AZURE_PLAIN_TEXT_KEY') }, 'layout' : { 'endpoint' : os.environ.get('AZURE_LAYOUT_ENDPOINT', ''), 'key' : os.environ.get('AZURE_LAYOUT_KEY', ''), 'model' : os.environ.get('AZURE_LAYOUT_MODEL', '') } } } GPT_ENGINE = 'o1-mini' GPT_KEY = os.environ.get('GPT_KEY', '') GPT_VERSION = '2024-12-01-preview' GPT_API_BASE = 'https://ackotest.openai.azure.com/' # EXTRACTION_PROMPT = open(os.path.join(PROMPTS_DIR, 'extraction.txt')).read() # entities = json.load(open(os.path.join(DATA_DIR, 'policy_analyser_entities.json'))) # for entity in entities: # del entity['entityId'] # entities_str = '\n---\n'.join(['\n'.join([f'{k} : {v}' for k, v in entity.items()]) for entity in entities]) # EXTRACTION_PROMPT += entities_str