wu981526092 commited on
Commit
6050414
1 Parent(s): 51b305a
Files changed (1) hide show
  1. model.py +5 -5
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=os.environ.get("AZURE_OPENAI_DEPLOYMENT_ID_RAG"),
13
- api_key=os.environ.get("OPENAI_API_KEY_RAG"),
14
- azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT_RAG"),
15
  api_version="2024-02-01",
16
  )
17
  embed_model = AzureOpenAIEmbedding(
18
  deployment_name="personality_rag_embedding",
19
- api_key=os.environ.get("OPENAI_API_KEY_RAG"),
20
- azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT_RAG"),
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