OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-11 08:01:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added valves to RAG pipe

Browse Source
This commit is contained in:
0xThresh.eth 2024-07-22 22:24:39 -06:00
parent 7b257d3a5c
commit 8401f02c91
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
examples/pipelines/rag/text_to_sql_pipeline.py
View File

@ -38,15 +38,15 @@ class Pipeline:
# Initialize
self.valves = self.Valves(
**{
"pipelines": ["*"], # Connect to all pipelines
"DB_HOST": os.environ["PG_HOST"], # Database hostname
"DB_PORT": os.environ["PG_PORT"], # Database port
"DB_USER": os.environ["PG_USER"], # User to connect to the database with
"DB_PASSWORD": os.environ["PG_PASSWORD"], # Password to connect to the database with
"DB_DATABASE": os.environ["PG_DB"], # Database to select on the DB instance
"DB_TABLES": ["albums"], # Table(s) to run queries against
"OLLAMA_HOST": "http://host.docker.internal:11434", # Make sure to update with the URL of your Ollama host, such as http://localhost:11434 or remote server address
"TEXT_TO_SQL_MODEL": "phi3:latest" # Model to use for text-to-SQL generation
"pipelines": ["*"], # Connect to all pipelines
"DB_HOST": os.getenv("PG_HOST", "http://localhost:5432"), # Database hostname
"DB_PORT": os.getenv("PG_PORT", 5432), # Database port
"DB_USER": os.getenv("PG_USER", "postgres"), # User to connect to the database with
"DB_PASSWORD": os.getenv("PG_PASSWORD", "password"), # Password to connect to the database with
"DB_DATABASE": os.getenv("PG_DB", "postgres"), # Database to select on the DB instance
"DB_TABLES": ["albums"], # Table(s) to run queries against
"OLLAMA_HOST": os.getenv("OLLAMA_HOST", "http://host.docker.internal:11434"), # Make sure to update with the URL of your Ollama host, such as http://localhost:11434 or remote server address
"TEXT_TO_SQL_MODEL": "phi3:latest" # Model to use for text-to-SQL generation
}
)