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

This commit is contained in:
0xThresh.eth 2024-07-02 13:45:09 -07:00
commit 7b257d3a5c
2 changed files with 8 additions and 2 deletions

View File

@ -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 @@ class Pipeline:
"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',

View File

@ -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