refactor: move import statement inside function to avoid unnecessary imports (#3505)
* fix: Add exception handling for missing langchain_anthropic package * refactor: Move import statement for get_starter_projects_dump in get_starter_projects * chore: Modify logger to use error level for exception in directory_reader
This commit is contained in:
parent
b5116e6f68
commit
2eca4d9e32
3 changed files with 10 additions and 5 deletions
|
|
@ -4,10 +4,8 @@ from fastapi import APIRouter, Depends, HTTPException
|
|||
from loguru import logger
|
||||
|
||||
from langflow.graph.graph.schema import GraphDump
|
||||
from langflow.services.database.models.user.model import User
|
||||
from langflow.services.auth.utils import get_current_active_user
|
||||
from langflow.initial_setup.load import get_starter_projects_dump
|
||||
|
||||
from langflow.services.database.models.user.model import User
|
||||
|
||||
router = APIRouter(prefix="/starter-projects", tags=["Flows"])
|
||||
|
||||
|
|
@ -18,6 +16,8 @@ def get_starter_projects(
|
|||
current_user: User = Depends(get_current_active_user),
|
||||
):
|
||||
"""Get a list of starter projects."""
|
||||
from langflow.initial_setup.load import get_starter_projects_dump
|
||||
|
||||
try:
|
||||
flows = get_starter_projects_dump()
|
||||
return flows
|
||||
|
|
|
|||
|
|
@ -1,4 +1,3 @@
|
|||
from langchain_anthropic.chat_models import ChatAnthropic
|
||||
from pydantic.v1 import SecretStr
|
||||
|
||||
from langflow.base.models.model import LCModelComponent
|
||||
|
|
@ -53,6 +52,12 @@ class AnthropicModelComponent(LCModelComponent):
|
|||
]
|
||||
|
||||
def build_model(self) -> LanguageModel: # type: ignore[type-var]
|
||||
try:
|
||||
from langchain_anthropic.chat_models import ChatAnthropic
|
||||
except ImportError:
|
||||
raise ImportError(
|
||||
"langchain_anthropic is not installed. Please install it with `pip install langchain_anthropic`."
|
||||
)
|
||||
model = self.model
|
||||
anthropic_api_key = self.anthropic_api_key
|
||||
max_tokens = self.max_tokens
|
||||
|
|
|
|||
|
|
@ -348,7 +348,7 @@ class DirectoryReader:
|
|||
try:
|
||||
output_types = await self.get_output_types_from_code_async(result_content)
|
||||
except Exception as exc:
|
||||
logger.exception(f"Error while getting output types from code: {str(exc)}")
|
||||
logger.error(f"Error while getting output types from code: {str(exc)}")
|
||||
output_types = [component_name_camelcase]
|
||||
else:
|
||||
output_types = [component_name_camelcase]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue