Merge branch 'main' into feat-text-to-sql-valves

open-webui · Jul 2, 2024 · 7b257d3 · 7b257d3
2 parents d5febd7 + a5bf4bd
commit 7b257d3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/examples/pipelines/providers/azure_openai_pipeline.py b/examples/pipelines/providers/azure_openai_pipeline.py
@@ -1,6 +1,7 @@
 from typing import List, Union, Generator, Iterator
 from pydantic import BaseModel
 import requests
+import os
 
 
 class Pipeline:
@@ -52,7 +53,7 @@ def pipe(
             "Content-Type": "application/json",
         }
 
-        url = f"{self.valves.AZURE_OPENAI_ENDPOINT}/openai/deployments/{self.valves.DEPLOYMENT_NAME}/chat/completions?api-version={self.valves.API_VERSION}"
+        url = f"{self.valves.AZURE_OPENAI_ENDPOINT}/openai/deployments/{self.valves.AZURE_OPENAI_DEPLOYMENT_NAME}/chat/completions?api-version={self.valves.AZURE_OPENAI_API_VERSION}"
 
         allowed_params = {'messages', 'temperature', 'role', 'content', 'contentPart', 'contentPartImage',
                           'enhancements', 'dataSources', 'n', 'stream', 'stop', 'max_tokens', 'presence_penalty',

diff --git a/requirements.txt b/requirements.txt
@@ -26,6 +26,7 @@ boto3
 redis
 sqlmodel
 chromadb
+psycopg2-binary
 
 # Observability
 langfuse
@@ -57,4 +58,8 @@ seaborn
 # Web scraping
 selenium
 playwright
-beautifulsoup4
+beautifulsoup4
+
+# Llama Index for RAG
+llama-index
+llama-index-llms-ollama