pieter/eveAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

- Add support for environment in communication with portkey

Browse Source
This commit is contained in:
Josako
2024-08-19 12:06:10 +02:00
parent 08ccb6170b
commit 926a4e8cc2
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.idea/misc.xml generated
View File

@@ -1,7 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?> <?xml version="1.0" encoding="UTF-8"?>
<project version="4"> <project version="4">
<component name="Black"> <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12 (eveAI)" project-jdk-type="Python SDK" />
<option name="sdkName" value="Python 3.12 (eveAI)" />
</component>
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12 (eveai_dev)" project-jdk-type="Python SDK" />
</project> </project>

2
.idea/modules.xml generated
View File

@@ -2,7 +2,7 @@
<project version="4"> <project version="4">
<component name="ProjectModuleManager"> <component name="ProjectModuleManager">
<modules> <modules>
<module fileurl="file://$PROJECT_DIR$/.idea/eveAI.iml" filepath="$PROJECT_DIR$/.idea/eveAI.iml" /> <module fileurl="file://$PROJECT_DIR$/.idea/TBD.iml" filepath="$PROJECT_DIR$/.idea/TBD.iml" />
</modules> </modules>
</component> </component>
</project> </project>

2
.python-version
View File

@@ -1 +1 @@
eveai_dev eveai_tbd

7
common/utils/model_utils.py
View File

@@ -1,3 +1,5 @@
import os
import langcodes import langcodes
from flask import current_app from flask import current_app
from langchain_openai import OpenAIEmbeddings, ChatOpenAI 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['min_chunk_size'] = tenant.min_chunk_size
model_variables['max_chunk_size'] = tenant.max_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 # Set Embedding variables
match embedding_provider: match embedding_provider:
case 'openai': case 'openai':
portkey_metadata = {'tenant_id': str(tenant.id)}
portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'), portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'),
provider='openai', provider='openai',
metadata=portkey_metadata) metadata=portkey_metadata)
@@ -123,7 +127,6 @@ def select_model_variables(tenant):
# Set Chat model variables # Set Chat model variables
match llm_provider: match llm_provider:
case 'openai': case 'openai':
portkey_metadata = {'tenant_id': str(tenant.id)}
portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'), portkey_headers = createHeaders(api_key=current_app.config.get('PORTKEY_API_KEY'),
metadata=portkey_metadata, metadata=portkey_metadata,
provider='openai') provider='openai')