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

ruff

Browse source
This commit is contained in:
hajdul88 2026-01-15 15:21:24 +01:00
parent 8dc358da39
commit 7ebe8563c5
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cognee/shared/usage_logger.py
View file

@ -288,12 +288,13 @@ if __name__ == "__main__":
self.id = user_id self.id = user_id
@log_usage(function_name="example_with_user", log_type="example") @log_usage(function_name="example_with_user", log_type="example")
async def example_with_user(data: str, user: MockUser, wrong_param=datetime.utcnow().isoformat()): async def example_with_user(
data: str, user: MockUser, wrong_param=datetime.utcnow().isoformat()
):
"""Example function with user parameter.""" """Example function with user parameter."""
await asyncio.sleep(0.05) await asyncio.sleep(0.05)
return float("nan") return float("nan")
@log_usage(function_name="returns_cycle", log_type="function") @log_usage(function_name="returns_cycle", log_type="function")
async def returns_cycle(): async def returns_cycle():
a = [] a = []
@ -314,7 +315,9 @@ if __name__ == "__main__":
# Example 2: Function with user # Example 2: Function with user
print("\n2. Running example function with user:") print("\n2. Running example function with user:")
mock_user = MockUser("example-user-123") mock_user = MockUser("example-user-123")
result2 = await example_with_user("sample_data", user=mock_user, wrong_param=datetime.utcnow().isoformat()) result2 = await example_with_user(
"sample_data", user=mock_user, wrong_param=datetime.utcnow().isoformat()
)
result3 = await example_with_user("sample_data", user=mock_user) result3 = await example_with_user("sample_data", user=mock_user)
print(f" Result: {result2}") print(f" Result: {result2}")
@ -353,7 +356,5 @@ if __name__ == "__main__":
print(" - CACHE_HOST=localhost") print(" - CACHE_HOST=localhost")
print(" - CACHE_PORT=6379") print(" - CACHE_PORT=6379")
# Run the example # Run the example
asyncio.run(run_example()) asyncio.run(run_example())