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

fix: initialize request object, correct typo, and rename parameter

Browse Source 
- Initialize `r` to `None` to prevent potential `NameError`.
- Correct typo: change "funcions" to "functions" in allowed parameters.
- Rename "dataSources" to "data_sources" to align with API specs.
This commit is contained in:
kmishmael 2024-09-01 18:52:09 +03:00
parent 68ded857a7
commit 0303263197
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/pipelines/providers/azure_openai_pipeline.py
View File

@ -56,8 +56,8 @@ class Pipeline:
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}" 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', allowed_params = {'messages', 'temperature', 'role', 'content', 'contentPart', 'contentPartImage',
'enhancements', 'dataSources', 'n', 'stream', 'stop', 'max_tokens', 'presence_penalty', 'enhancements', 'data_sources', 'n', 'stream', 'stop', 'max_tokens', 'presence_penalty',
'frequency_penalty', 'logit_bias', 'user', 'function_call', 'funcions', 'tools', 'frequency_penalty', 'logit_bias', 'user', 'function_call', 'functions', 'tools',
'tool_choice', 'top_p', 'log_probs', 'top_logprobs', 'response_format', 'seed'} 'tool_choice', 'top_p', 'log_probs', 'top_logprobs', 'response_format', 'seed'}
# remap user field # remap user field
if "user" in body and not isinstance(body["user"], str): if "user" in body and not isinstance(body["user"], str):
@ -67,6 +67,8 @@ class Pipeline:
if len(body) != len(filtered_body): if len(body) != len(filtered_body):
print(f"Dropped params: {', '.join(set(body.keys()) - set(filtered_body.keys()))}") print(f"Dropped params: {', '.join(set(body.keys()) - set(filtered_body.keys()))}")
# Initialize the response variable to None.
r = None
try: try:
r = requests.post( r = requests.post(
url=url, url=url,