wehub-resource-sync
c889a57b6b
Test Suites / Build CI Environment (push) Has been cancelled
Test Suites / Basic Tests (push) Has been cancelled
Test Suites / End-to-End Tests (push) Has been cancelled
Test Suites / CLI Tests (push) Has been cancelled
Test Suites / Slow End-to-End Tests (push) Has been cancelled
Test Suites / Graph Database Tests (push) Has been cancelled
Test Suites / Vector DB Tests (push) Has been cancelled
Test Suites / Temporal Graph Test (push) Has been cancelled
Test Suites / Search Test on Different DBs (push) Has been cancelled
Test Suites / Example Tests (push) Has been cancelled
Test Suites / Notebook Tests (push) Has been cancelled
Test Suites / OS and Python Tests Ubuntu (push) Has been cancelled
Test Suites / OS and Python Tests Extended (push) Has been cancelled
Test Suites / LLM Test Suite (push) Has been cancelled
Test Suites / S3 File Storage Test (push) Has been cancelled
Test Suites / Run Integration Tests (push) Has been cancelled
Test Suites / MCP Tests (push) Has been cancelled
Test Suites / Docker Compose Test (push) Has been cancelled
Test Suites / Docker CI test (push) Has been cancelled
Test Suites / Relational DB Migration Tests (push) Has been cancelled
Test Suites / Distributed Cognee Test (push) Has been cancelled
Test Suites / DB Examples Tests (push) Has been cancelled
Test Suites / Test Completion Status (push) Has been cancelled
Test Suites / Claude Code Review (push) Has been cancelled
Test Suites / basic checks (push) Has been cancelled
build | Build and Push Cognee MCP Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
build | Build and Push Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.11) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.12) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (kuzu, kuzu) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (neo4j, neo4j) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Examples (push) Has been cancelled
Weighted Edges Tests / Code Quality for Weighted Edges (push) Has been cancelled
47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from cognee.modules.engine.models import Interval, Event
|
|
from cognee.modules.engine.utils.generate_timestamp_datapoint import generate_timestamp_datapoint
|
|
|
|
|
|
def generate_event_datapoint(event) -> Event:
|
|
"""
|
|
Generates an Event datapoint from a given event model, including temporal metadata if available.
|
|
|
|
The function maps the basic attributes (name, description, location) from the input event
|
|
and enriches them with temporal information. If start and end times are provided, an
|
|
Interval is created. If only one timestamp is available, it is added directly. Temporal
|
|
information is also appended to the event description for context.
|
|
|
|
Args:
|
|
event: An event model instance containing attributes such as name, description,
|
|
location, time_from, and time_to.
|
|
|
|
Returns:
|
|
Event: A structured Event object with name, description, location, and enriched
|
|
temporal details.
|
|
"""
|
|
# Base event data
|
|
event_data = {
|
|
"name": event.name,
|
|
"description": event.description,
|
|
"location": event.location,
|
|
}
|
|
|
|
# Create timestamps if they exist
|
|
time_from = generate_timestamp_datapoint(event.time_from) if event.time_from else None
|
|
time_to = generate_timestamp_datapoint(event.time_to) if event.time_to else None
|
|
|
|
# Add temporal information
|
|
if time_from and time_to:
|
|
event_data["during"] = Interval(time_from=time_from, time_to=time_to)
|
|
# Enrich description with temporal info
|
|
temporal_info = f"\n---\nTime data: {time_from.timestamp_str} to {time_to.timestamp_str}"
|
|
event_data["description"] = (event_data["description"] or "Event") + temporal_info
|
|
elif time_from or time_to:
|
|
timestamp = time_from or time_to
|
|
event_data["at"] = timestamp
|
|
# Enrich description with temporal info
|
|
temporal_info = f"\n---\nTime data: {timestamp.timestamp_str}"
|
|
event_data["description"] = (event_data["description"] or "Event") + temporal_info
|
|
|
|
return Event(**event_data)
|