Merge pull request #31 from GrayCodeAI/feat/cognee-composio-adoption

feat: add composio/cognee SDK methods

Author: Patel230

src/hawk/__init__.py

@@ -3,6 +3,26 @@
 from ._version import __version__
 from .agent import Agent, AgentConfig, AsyncAgent
 from .client import AsyncHawkClient, HawkClient
+from .composio_cognee import (
+    CogneeFeedbackEntry,
+    CogneeImproveResult,
+    CogneeQAEntry,
+    CogneeSkillRunEntry,
+    CogneeTraceEntry,
+    ComposioCredential,
+    ComposioTool,
+    ComposioToolResult,
+    execute_composio_tool,
+    improve,
+    list_composio_credentials,
+    recall_with_session,
+    remember_feedback,
+    remember_qa,
+    remember_skill_run,
+    remember_trace,
+    search_composio_tools,
+    sync_session_to_permanent,
+)
 from .discovery import (
     AgentCard,
     AgentResolver,
@@ -89,6 +109,14 @@
     "BenchmarkResults",
     "ChatRequest",
     "ChatResponse",
+    "CogneeFeedbackEntry",
+    "CogneeImproveResult",
+    "CogneeQAEntry",
+    "CogneeSkillRunEntry",
+    "CogneeTraceEntry",
+    "ComposioCredential",
+    "ComposioTool",
+    "ComposioToolResult",
     "CompositeResolver",
     "EvalResult",
     "EvalTask",
@@ -141,9 +169,19 @@
     "chat_with_tools_async",
     "configure_tracing",
     "detect_provider",
+    "execute_composio_tool",
+    "improve",
     "is_tracing_enabled",
+    "list_composio_credentials",
+    "recall_with_session",
+    "remember_feedback",
+    "remember_qa",
+    "remember_skill_run",
+    "remember_trace",
     "run_benchmark",
     "run_benchmark_async",
+    "search_composio_tools",
+    "sync_session_to_permanent",
     "tool",
     "trace",
     "trace_chat",