github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-01-24 05:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
Lance Martin
2023-11-10 14:18:32 -08:00
parent 355eba2b6b
commit a6a562955f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/llms/ollama.py
View File

@@ -97,6 +97,7 @@ class _OllamaCommon(BaseLanguageModel):
"""Get the default parameters for calling Ollama."""
return {
"model": self.model,
"format": self.format,
"options": {
"mirostat": self.mirostat,
"mirostat_eta": self.mirostat_eta,
@@ -111,14 +112,13 @@ class _OllamaCommon(BaseLanguageModel):
"tfs_z": self.tfs_z,
"top_k": self.top_k,
"top_p": self.top_p,
"format": self.format,
},
}
@property
def _identifying_params(self) -> Mapping[str, Any]:
"""Get the identifying parameters."""
return {**{"model": self.model}, **self._default_params}
return {**{"model": self.model,"format": self.format}, **self._default_params}
def _create_stream(
self,