Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-20 03:25:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Ollama - Pass headers to post request in async method (#16660)

Browse Source 
# Description
A previous PR (https://github.com/langchain-ai/langchain/pull/15881)
added option to pass headers to ollama endpoint, but headers are not
pass to the async method.
This commit is contained in:
Zhuoyun(John) Xu 2024-01-27 17:11:32 -07:00 committed by GitHub
parent 5e73603e8a
commit 508bde7f40
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/langchain_community/llms/ollama.py
View File

@ -284,7 +284,10 @@ class _OllamaCommon(BaseLanguageModel):
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.post( async with session.post(
url=api_url, url=api_url,
headers={"Content-Type": "application/json"}, headers={
"Content-Type": "application/json",
**(self.headers if isinstance(self.headers, dict) else {}),
},
json=request_payload, json=request_payload,
timeout=self.timeout, timeout=self.timeout,
) as response: ) as response: