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

Update cognee-mcp/cognee_mcp/server.py

Browse source 
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Vasilije 2025-01-11 16:57:22 +01:00 • committed by GitHub
parent daf2d540ba
commit b132ff4990
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
cognee-mcp/cognee_mcp/server.py
View file

@ -103,6 +103,9 @@ async def handle_list_tools() -> list[types.Tool]:
def get_freshest_png(directory: str) -> Image.Image: def get_freshest_png(directory: str) -> Image.Image:
if not os.path.exists(directory):
raise FileNotFoundError(f"Directory {directory} does not exist")
# List all files in 'directory' that end with .png # List all files in 'directory' that end with .png
files = [f for f in os.listdir(directory) if f.endswith(".png")] files = [f for f in os.listdir(directory) if f.endswith(".png")]
if not files: if not files:
@ -110,14 +113,20 @@ def get_freshest_png(directory: str) -> Image.Image:
# Sort by integer value of the filename (minus the '.png') # Sort by integer value of the filename (minus the '.png')
# Example filename: 1673185134.png -> integer 1673185134 # Example filename: 1673185134.png -> integer 1673185134
files_sorted = sorted(files, key=lambda x: int(x.replace(".png", ""))) try:
files_sorted = sorted(files, key=lambda x: int(x.replace(".png", "")))
except ValueError as e:
raise ValueError("Invalid PNG filename format. Expected timestamp format.") from e
# The "freshest" file has the largest timestamp # The "freshest" file has the largest timestamp
freshest_filename = files_sorted[-1] freshest_filename = files_sorted[-1]
freshest_path = os.path.join(directory, freshest_filename) freshest_path = os.path.join(directory, freshest_filename)
# Open the image with PIL and return the PIL Image object # Open the image with PIL and return the PIL Image object
return Image.open(freshest_path) try:
return Image.open(freshest_path)
except (IOError, OSError) as e:
raise IOError(f"Failed to open PNG file {freshest_path}") from e
@server.call_tool() @server.call_tool()
async def handle_call_tool( async def handle_call_tool(