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

fix: changed import and linting

Browse source
This commit is contained in:
Gabriel Almeida 2023-03-20 20:56:15 -03:00
commit 049ab2b39b
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/langflow/__main__.py
View file

@ -1,6 +1,5 @@
import multiprocessing import multiprocessing
import platform import platform
import re
from langflow.main import create_app from langflow.main import create_app

6
src/backend/langflow/utils/util.py
View file

@ -11,7 +11,7 @@ from langchain.agents.load_tools import (
) )
from typing import Optional, Dict from typing import Optional, Dict
from backend.langflow.utils.constants import CHAT_OPENAI_MODELS, OPENAI_MODELS from langflow.utils import constants
def build_template_from_function(name: str, type_to_loader_dict: Dict): def build_template_from_function(name: str, type_to_loader_dict: Dict):
@ -298,9 +298,9 @@ def format_dict(d, name: Optional[str] = None):
# Add options to openai # Add options to openai
if name == "OpenAI" and key == "model_name": if name == "OpenAI" and key == "model_name":
value["options"] = OPENAI_MODELS value["options"] = constants.OPENAI_MODELS
elif name == "OpenAIChat" and key == "model_name": elif name == "OpenAIChat" and key == "model_name":
value["options"] = CHAT_OPENAI_MODELS value["options"] = constants.CHAT_OPENAI_MODELS
return d return d