add text_llm in rag_1

artefactory · Mar 22, 2024 · be651f3 · be651f3
1 parent 6e0f384
commit be651f3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/backend/rag_1/config.py b/backend/rag_1/config.py
@@ -72,6 +72,7 @@ class Config:
 
     path: PathConfig
 
+    text_llm: HydraObject
     vision_llm: HydraObject
     embedding: HydraObject
     vectorstore: HydraObject

diff --git a/backend/rag_1/config.yaml b/backend/rag_1/config.yaml
@@ -5,6 +5,15 @@ path:
   export_extracted: "${.docs}/extracted/${..name}"
   database: "database/${..name}"
 
+text_llm:
+  _target_: langchain_openai.AzureChatOpenAI
+  azure_endpoint: ${oc.env:TEXT_OPENAI_ENDPOINT}
+  openai_api_key: ${oc.env:TEXT_OPENAI_API_KEY}
+  openai_api_version: "2024-02-15-preview"
+  deployment_name: "gpt4-turbo"
+  temperature: 0.0
+  max_tokens: 1024
+
 vision_llm:
   _target_: langchain_openai.AzureChatOpenAI
   azure_endpoint: ${oc.env:VISION_OPENAI_ENDPOINT}