wu981526092
commited on
Commit
•
6050414
1
Parent(s):
51b305a
add
Browse files
model.py
CHANGED
@@ -9,15 +9,15 @@ import os
|
|
9 |
# Create LLM and Embedding models
|
10 |
def create_models():
|
11 |
llm = AzureOpenAI(
|
12 |
-
deployment_name=
|
13 |
-
api_key=os.environ.get("
|
14 |
-
azure_endpoint=
|
15 |
api_version="2024-02-01",
|
16 |
)
|
17 |
embed_model = AzureOpenAIEmbedding(
|
18 |
deployment_name="personality_rag_embedding",
|
19 |
-
api_key=os.environ.get("
|
20 |
-
azure_endpoint=
|
21 |
api_version="2024-02-01",
|
22 |
)
|
23 |
return llm, embed_model
|
|
|
9 |
# Create LLM and Embedding models
|
10 |
def create_models():
|
11 |
llm = AzureOpenAI(
|
12 |
+
deployment_name="personality_gpt4o",
|
13 |
+
api_key=os.environ.get("AZURE_OPENAI_KEY"),
|
14 |
+
azure_endpoint="https://personalityanalysisfinetuning.openai.azure.com/",
|
15 |
api_version="2024-02-01",
|
16 |
)
|
17 |
embed_model = AzureOpenAIEmbedding(
|
18 |
deployment_name="personality_rag_embedding",
|
19 |
+
api_key=os.environ.get("AZURE_OPENAI_KEY"),
|
20 |
+
azure_endpoint="https://personalityanalysisfinetuning.openai.azure.com/",
|
21 |
api_version="2024-02-01",
|
22 |
)
|
23 |
return llm, embed_model
|