Update dev

This commit is contained in:
William Fu-Hinthorn
2024-12-13 17:01:01 -08:00
parent dbf5b4920e
commit 70f2efc9a1
3 changed files with 14 additions and 9 deletions
+11 -8
View File
@@ -576,16 +576,18 @@ def dev(
from langgraph_api.cli import run_server
except ImportError:
try:
import pkg_resources
from importlib import util
pkg_resources.require("langgraph-api-inmem")
except (ImportError, pkg_resources.DistributionNotFound):
if not util.find_spec("langgraph_api"):
raise click.UsageError(
"Required package 'langgraph-api' is not installed.\n"
"Please install it with:\n\n"
' pip install -U "langgraph-cli[inmem]"\n\n'
) from None
except ImportError:
raise click.UsageError(
"Required package 'langgraph-api-inmem' is not installed.\n"
"Please install it with:\n\n"
' pip install -U "langgraph-cli[inmem]"\n\n'
"If you're developing the langgraph-cli package locally, you can install in development mode:\n"
" pip install -e ."
"Could not verify package installation. Please ensure Python is up to date and\n"
"langgraph-cli is installed with the 'inmem' extra: pip install -U \"langgraph-cli[inmem]\""
) from None
raise click.UsageError(
"Could not import run_server. This likely means your installation is incomplete.\n"
@@ -614,6 +616,7 @@ def dev(
env=config_json.get("env"),
store=config_json.get("store"),
wait_for_client=wait_for_client,
auth=config_json.get("auth"),
)
+1 -1
View File
@@ -61,7 +61,7 @@ class Config(TypedDict, total=False):
graphs: dict[str, str]
env: Union[dict[str, str], str]
store: Optional[StoreConfig]
auth: Optional[dict[str, str]]
auth: Optional[AuthConfig]
def _parse_version(version_str: str) -> tuple[int, int]:
+2
View File
@@ -31,6 +31,7 @@ def test_validate_config():
"dockerfile_lines": [],
"env": {},
"store": None,
"auth": None,
**expected_config,
}
actual_config = validate_config(expected_config)
@@ -48,6 +49,7 @@ def test_validate_config():
},
"env": env,
"store": None,
"auth": None,
}
actual_config = validate_config(expected_config)
assert actual_config == expected_config