ollama[patch]: delete suffix slash to avoid redirect (#18260)

- **Description:** see
[ollama](https://github.com/ollama/ollama/blob/main/server/routes.go#L949)'s
route definitions
- **Issue:** N/A
- **Dependencies:** N/A
This commit is contained in:
mackong 2024-02-29 05:44:48 +08:00 committed by GitHub
parent 6b58943917
commit 2c42f3a955
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -159,7 +159,7 @@ class ChatOllama(BaseChatModel, _OllamaCommon):
"messages": self._convert_messages_to_ollama_messages(messages), "messages": self._convert_messages_to_ollama_messages(messages),
} }
yield from self._create_stream( yield from self._create_stream(
payload=payload, stop=stop, api_url=f"{self.base_url}/api/chat/", **kwargs payload=payload, stop=stop, api_url=f"{self.base_url}/api/chat", **kwargs
) )
async def _acreate_chat_stream( async def _acreate_chat_stream(
@ -172,7 +172,7 @@ class ChatOllama(BaseChatModel, _OllamaCommon):
"messages": self._convert_messages_to_ollama_messages(messages), "messages": self._convert_messages_to_ollama_messages(messages),
} }
async for stream_resp in self._acreate_stream( async for stream_resp in self._acreate_stream(
payload=payload, stop=stop, api_url=f"{self.base_url}/api/chat/", **kwargs payload=payload, stop=stop, api_url=f"{self.base_url}/api/chat", **kwargs
): ):
yield stream_resp yield stream_resp