gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

temp

Browse source
This commit is contained in:
Daulet Amirkhanov 2025-10-13 15:54:25 +01:00
parent 9cc941a0be
commit fa338bb7ab
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cognee/tests/cli_tests/cli_integration_tests/test_cli_integration.py
View file

@ -87,17 +87,33 @@ class TestCliIntegration:
# Note: This might fail due to dependencies, but we're testing the CLI structure # Note: This might fail due to dependencies, but we're testing the CLI structure
# The important thing is that it doesn't crash with argument parsing errors # The important thing is that it doesn't crash with argument parsing errors
# Allow litellm logging worker cancellation errors as they're expected during process shutdown # Allow litellm logging worker cancellation errors as they're expected during process shutdown
# Also allow Kuzu extension errors as they're non-critical warnings in CI environments
stderr_lower = result.stderr.lower() stderr_lower = result.stderr.lower()
has_error = "error" in stderr_lower has_error = "error" in stderr_lower
has_expected_failure = "failed to add data" in stderr_lower has_expected_failure = "failed to add data" in stderr_lower
has_litellm_cancellation = ( has_litellm_cancellation = (
"loggingworker cancelled" in stderr_lower or "cancellederror" in stderr_lower "loggingworker cancelled" in stderr_lower or "cancellederror" in stderr_lower
) )
has_kuzu_extension_error = (
"could not establish connection" in stderr_lower
and "extension.kuzudb.com" in stderr_lower
) or (
"binder exception: extension" in stderr_lower
and "has not been installed" in stderr_lower
)
try: try:
assert not has_error or has_expected_failure or has_litellm_cancellation assert (
not has_error
or has_expected_failure
or has_litellm_cancellation
or has_kuzu_extension_error
)
except Exception as e: except Exception as e:
logger.error(f"{str(e)}", exc_info=True) logger.error(f"{str(e)}", exc_info=True)
import pprint
pprint.pprint(result.stderr)
raise e raise e
finally: finally: