bb68d6a0df
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
51 lines
2 KiB
Python
51 lines
2 KiB
Python
from cognee.shared.data_models import SummarizedCode, SummarizedClass, SummarizedFunction
|
|
|
|
|
|
def get_mock_summarized_code() -> SummarizedCode:
|
|
"""
|
|
Return a summarized representation of mock code.
|
|
|
|
This function constructs and returns a `SummarizedCode` object that includes various
|
|
components such as file name, high-level summary, key features, imports, constants,
|
|
classes, and functions, all described with placeholders for mock data.
|
|
|
|
Returns:
|
|
--------
|
|
|
|
- SummarizedCode: A `SummarizedCode` object containing mock data for file summary,
|
|
features, imports, constants, classes, and functions.
|
|
"""
|
|
return SummarizedCode(
|
|
file_name="mock_file.py",
|
|
high_level_summary="This is a mock high-level summary.",
|
|
key_features=["Mock feature 1", "Mock feature 2"],
|
|
imports=["mock_import1", "mock_import2"],
|
|
constants=["MOCK_CONSTANT = 'mock_value'"],
|
|
classes=[
|
|
SummarizedClass(
|
|
name="MockClass",
|
|
description="This is a mock description of the MockClass.",
|
|
methods=[
|
|
SummarizedFunction(
|
|
name="mock_method",
|
|
description="This is a description of the mock method.",
|
|
docstring="This is a mock method.",
|
|
inputs=["mock_input: str"],
|
|
outputs=["mock_output: str"],
|
|
decorators=None,
|
|
)
|
|
],
|
|
)
|
|
],
|
|
functions=[
|
|
SummarizedFunction(
|
|
name="mock_function",
|
|
description="This is a description of the mock function.",
|
|
docstring="This is a mock function.",
|
|
inputs=["mock_input: str"],
|
|
outputs=["mock_output: str"],
|
|
decorators=None,
|
|
)
|
|
],
|
|
workflow_description="This is a mock workflow description.",
|
|
)
|