jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor OllamaLLM imports and remove unnecessary code

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-07 20:06:20 -03:00
commit de292b8fff
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/components/llms/OllamaLLM.py
View file

@ -1,8 +1,7 @@
from typing import Optional, List from typing import List, Optional
from langchain.llms import Ollama
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from langchain_community.llms.ollama import Ollama
from langflow import CustomComponent from langflow import CustomComponent
@ -151,10 +150,9 @@ class OllamaLLM(CustomComponent):
"top_p": top_p, "top_p": top_p,
} }
# None Value remove # None Value remove
llm_params = {k: v for k, v in llm_params.items() if v is not None} llm_params = {k: v for k, v in llm_params.items() if v is not None}
try: try:
llm = Ollama(**llm_params) llm = Ollama(**llm_params)
except Exception as e: except Exception as e: