diff --git a/cognee-frontend/src/app/(graph)/CrewAITrigger.tsx b/cognee-frontend/src/app/(graph)/CrewAITrigger.tsx index 8252e3f34..0c3625ff3 100644 --- a/cognee-frontend/src/app/(graph)/CrewAITrigger.tsx +++ b/cognee-frontend/src/app/(graph)/CrewAITrigger.tsx @@ -82,6 +82,7 @@ export default function CrewAITrigger({ onData, onActivity }: CrewAITriggerProps .then(response => response.json()) .then(() => { onActivity([{ id: uuid4(), timestamp: Date.now(), activity: "Hiring crew agents made a decision" }]); + onData(null); }) .catch(() => { onActivity([{ id: uuid4(), timestamp: Date.now(), activity: "Hiring crew agents had problems while executing" }]); diff --git a/cognee-frontend/src/app/(graph)/GraphControls.tsx b/cognee-frontend/src/app/(graph)/GraphControls.tsx index 9133dbaba..4de65db43 100644 --- a/cognee-frontend/src/app/(graph)/GraphControls.tsx +++ b/cognee-frontend/src/app/(graph)/GraphControls.tsx @@ -87,7 +87,7 @@ export default function GraphControls({ isAddNodeFormOpen, onGraphShapeChange, o setTimeout(() => { onFitIntoView(); }, 500); - }); + }, [onFitIntoView, onGraphShapeChange]); return ( <> diff --git a/cognee-frontend/src/app/(graph)/GraphView.tsx b/cognee-frontend/src/app/(graph)/GraphView.tsx index 680a7bd77..b868cb3ea 100644 --- a/cognee-frontend/src/app/(graph)/GraphView.tsx +++ b/cognee-frontend/src/app/(graph)/GraphView.tsx @@ -35,6 +35,12 @@ export default function GraphView() { const [data, updateData] = useState(); const onDataChange = useCallback((newData: NodesAndEdges) => { + if (newData === null) { + // Requests for resetting the data + updateData(undefined); + return; + } + if (!newData.nodes.length && !newData.links.length) { return; } diff --git a/cognee-frontend/src/app/(graph)/getColorForNodeType.ts b/cognee-frontend/src/app/(graph)/getColorForNodeType.ts index 2ebda29c9..4edaa334d 100644 --- a/cognee-frontend/src/app/(graph)/getColorForNodeType.ts +++ b/cognee-frontend/src/app/(graph)/getColorForNodeType.ts @@ -2,9 +2,10 @@ import colors from "tailwindcss/colors"; import { formatHex } from "culori"; const NODE_COLORS = { - Document: formatHex(colors.blue[500]), - Chunk: formatHex(colors.green[500]), - Entity: formatHex(colors.yellow[500]), + TextDocument: formatHex(colors.blue[500]), + DocumentChunk: formatHex(colors.green[500]), + TextSummary: formatHex(colors.orange[500]), + Entity: formatHex(colors.yellow[300]), EntityType: formatHex(colors.purple[800]), NodeSet: formatHex(colors.indigo[300]), GitHubUser: formatHex(colors.gray[300]), diff --git a/cognee/complex_demos/crewai_demo/src/crewai_demo/hiring_crew.py b/cognee/complex_demos/crewai_demo/src/crewai_demo/hiring_crew.py index dda0308b0..8055eac50 100644 --- a/cognee/complex_demos/crewai_demo/src/crewai_demo/hiring_crew.py +++ b/cognee/complex_demos/crewai_demo/src/crewai_demo/hiring_crew.py @@ -25,7 +25,6 @@ class HiringCrew: def __init__(self, user, inputs): self.user = user self.inputs = inputs - self @agent def soft_skills_expert_agent(self) -> Agent: diff --git a/cognee/complex_demos/crewai_demo/src/crewai_demo/main.py b/cognee/complex_demos/crewai_demo/src/crewai_demo/main.py index 23389bdda..a8a806f63 100644 --- a/cognee/complex_demos/crewai_demo/src/crewai_demo/main.py +++ b/cognee/complex_demos/crewai_demo/src/crewai_demo/main.py @@ -1,4 +1,5 @@ import os +import random import time import warnings from uuid import uuid4 @@ -20,8 +21,18 @@ def print_environment(): print(f"{key}={os.environ[key]}") +github_tokens = [ + os.getenv("GITHUB_TOKEN"), + os.getenv("GITHUB_TOKEN_FALLBACK_1"), + os.getenv("GITHUB_TOKEN_FALLBACK_2"), + os.getenv("GITHUB_TOKEN_FALLBACK_3"), + os.getenv("GITHUB_TOKEN_FALLBACK_4"), + os.getenv("GITHUB_TOKEN_FALLBACK_5"), +] + + async def run_github_ingestion(user, dataset, applicant_1, applicant_2): - token = os.getenv("GITHUB_TOKEN") + token = random.choice(github_tokens) pipeline_run_id = get_crewai_pipeline_run_id(user.id)