diff --git a/.idea/misc.xml b/.idea/misc.xml index e8182a0..8136279 100644 --- a/.idea/misc.xml +++ b/.idea/misc.xml @@ -1,7 +1,4 @@ - - - + \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml index e9edca8..dab5cf5 100644 --- a/.idea/modules.xml +++ b/.idea/modules.xml @@ -2,7 +2,7 @@ - + \ No newline at end of file diff --git a/.python-version b/.python-version index af25bdb..ec2af43 100644 --- a/.python-version +++ b/.python-version @@ -1 +1 @@ -eveai_dev +eveai_tbd diff --git a/common/utils/model_utils.py b/common/utils/model_utils.py index 9470685..320dd99 100644 --- a/common/utils/model_utils.py +++ b/common/utils/model_utils.py @@ -1,3 +1,5 @@ +import os + import langcodes from flask import current_app from langchain_openai import OpenAIEmbeddings, ChatOpenAI @@ -89,10 +91,12 @@ def select_model_variables(tenant): model_variables['min_chunk_size'] = tenant.min_chunk_size model_variables['max_chunk_size'] = tenant.max_chunk_size + environment = os.getenv('FLASK_ENV', 'development') + portkey_metadata = {'tenant_id': str(tenant.id), 'environment': environment} + # Set Embedding variables match embedding_provider: case 'openai': - portkey_metadata = {'tenant_id': str(tenant.id)} portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'), provider='openai', metadata=portkey_metadata) @@ -123,7 +127,6 @@ def select_model_variables(tenant): # Set Chat model variables match llm_provider: case 'openai': - portkey_metadata = {'tenant_id': str(tenant.id)} portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'), metadata=portkey_metadata, provider='openai')