Spaces:
Running
Running
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 |