diff --git a/examples/pipelines/providers/azure_openai_pipeline.py b/examples/pipelines/providers/azure_openai_pipeline.py index aae75813..39146e00 100644 --- 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 index c5700a70..7b102545 100644 --- 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 \ No newline at end of file +beautifulsoup4 + +# Llama Index for RAG +llama-index +llama-index-llms-ollama \ No newline at end of file