gumloop 0.3.2__tar.gz → 0.3.3__tar.gz

{gumloop-0.3.2 → gumloop-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gumloop
-Version: 0.3.2
+Version: 0.3.3
 Summary: Python client for the Gumloop API
 Project-URL: Homepage, https://github.com/gumloop/gumloop-py
 Project-URL: Bug Tracker, https://github.com/gumloop/gumloop-py/issues

{gumloop-0.3.2 → gumloop-0.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "gumloop"
-version = "0.3.2"
+version = "0.3.3"
 description = "Python client for the Gumloop API"
 readme = "README.md"
 requires-python = ">=3.10"

{gumloop-0.3.2 → gumloop-0.3.3}/src/gumloop/__init__.py

@@ -6,7 +6,7 @@ from gumloop.errors import AuthenticationError
 from gumloop.errors import GumloopError
 from gumloop.oauth import OAuth
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"
 __all__ = [
     "APIStatusError",
     "AsyncGumloop",

{gumloop-0.3.2 → gumloop-0.3.3}/src/gumloop/_client.py

@@ -15,12 +15,14 @@ from gumloop.resources import Agents
 from gumloop.resources import Artifacts
 from gumloop.resources import AsyncAgents
 from gumloop.resources import AsyncArtifacts
+from gumloop.resources import AsyncBrain
 from gumloop.resources import AsyncChat
 from gumloop.resources import AsyncMCP
 from gumloop.resources import AsyncModels
 from gumloop.resources import AsyncSessions
 from gumloop.resources import AsyncSkills
 from gumloop.resources import AsyncTeams
+from gumloop.resources import Brain
 from gumloop.resources import Chat
 from gumloop.resources import Models
 from gumloop.resources import Sessions
@@ -41,9 +43,9 @@ def _derive_stream_base_url(base_url: str) -> str:
 class Gumloop:
     """Sync Gumloop SDK client. Composes an :class:`HttpClient` with the
     resource classes (``agents``, ``sessions``, ``mcp``, ``teams``,
-    ``skills``, ``artifacts``, ``models``). The :attr:`oauth` attribute
-    holds OAuth helpers — kept off the transport because it bootstraps the
-    bearer token rather than consuming one."""
+    ``skills``, ``artifacts``, ``brain``, ``models``). The :attr:`oauth`
+    attribute holds OAuth helpers — kept off the transport because it
+    bootstraps the bearer token rather than consuming one."""
 
     def __init__(
         self,
@@ -85,6 +87,7 @@ class Gumloop:
         self.teams = Teams(self._http)
         self.skills = Skills(self._http)
         self.artifacts = Artifacts(self._http)
+        self.brain = Brain(self._http)
         self.oauth = OAuth(base_url=self.base_url, timeout=self.timeout)
 
     def close(self) -> None:
@@ -137,6 +140,7 @@ class AsyncGumloop:
         self.teams = AsyncTeams(self._http)
         self.skills = AsyncSkills(self._http)
         self.artifacts = AsyncArtifacts(self._http)
+        self.brain = AsyncBrain(self._http)
         self.oauth = OAuth(base_url=self.base_url, timeout=self.timeout)
 
     async def __aenter__(self) -> AsyncGumloop:

gumloop-0.3.3/src/gumloop/cli/commands/brain.py

@@ -0,0 +1,66 @@
+from __future__ import annotations
+
+from typing import Annotated
+
+import typer
+
+from gumloop import GumloopError
+from gumloop.cli.console import console
+from gumloop.cli.console import print_json
+from gumloop.cli.context import CliContext
+from gumloop.cli.errors import exit_with_error
+
+brain_app = typer.Typer(help="Search your Company Brain.", no_args_is_help=True, rich_markup_mode="rich")
+
+
+@brain_app.command(
+    "search",
+    epilog=(
+        "Examples:\n"
+        '  gumloop brain search "onboarding process"\n'
+        '  gumloop brain search "pricing" --limit 5 --source notion --json'
+    ),
+)
+def search_brain(
+    ctx: typer.Context,
+    query: Annotated[str, typer.Argument(help="Search query.")],
+    limit: Annotated[
+        int | None,
+        typer.Option("--limit", help="Maximum number of results to return (1-50)."),
+    ] = None,
+    source: Annotated[
+        list[str] | None,
+        typer.Option("--source", help="Filter by source type (repeatable), e.g. notion, google_drive, slack."),
+    ] = None,
+    json_output: Annotated[
+        bool,
+        typer.Option("--json", help="Print the raw SDK response as JSON."),
+    ] = False,
+) -> None:
+    """Search indexed Company Brain sources."""
+    cli: CliContext = ctx.obj
+    try:
+        response = cli.call_with_refresh(
+            lambda client: client.brain.search(query, limit=limit, source_type=source or None)
+        )
+    except GumloopError as error:
+        exit_with_error(error, json_output=json_output)
+
+    if json_output:
+        print_json(response)
+        return
+
+    if not response.results:
+        console.print("No results found.")
+        return
+
+    console.print("SCORE", "SOURCE", "TITLE", "URL", sep="\t", soft_wrap=True)
+    for result in response.results:
+        console.print(
+            "" if result.score is None else f"{result.score:.3f}",
+            result.source or "",
+            result.title or "",
+            result.url or "",
+            sep="\t",
+            soft_wrap=True,
+        )

{gumloop-0.3.2 → gumloop-0.3.3}/src/gumloop/cli/main.py

@@ -11,6 +11,7 @@ from gumloop.cli.commands.agents import agents_app
 from gumloop.cli.commands.artifacts import artifacts_app
 from gumloop.cli.commands.auth import login as login_command
 from gumloop.cli.commands.auth import logout as logout_command
+from gumloop.cli.commands.brain import brain_app
 from gumloop.cli.commands.chat import chat_app
 from gumloop.cli.commands.mcp import mcp_app
 from gumloop.cli.commands.sessions import sessions_app
@@ -97,6 +98,7 @@ app.add_typer(sessions_app, name="sessions")
 app.add_typer(skills_app, name="skills")
 app.add_typer(artifacts_app, name="artifacts")
 app.add_typer(chat_app, name="chat")
+app.add_typer(brain_app, name="brain")
 
 
 def _require_supported_platform() -> None:

{gumloop-0.3.2 → gumloop-0.3.3}/src/gumloop/resources/__init__.py

@@ -6,6 +6,8 @@ from gumloop.resources.agents import AsyncModels
 from gumloop.resources.agents import Models
 from gumloop.resources.artifacts import Artifacts
 from gumloop.resources.artifacts import AsyncArtifacts
+from gumloop.resources.brain import AsyncBrain
+from gumloop.resources.brain import Brain
 from gumloop.resources.chat import AsyncChat
 from gumloop.resources.chat import Chat
 from gumloop.resources.mcp import MCP
@@ -23,12 +25,14 @@ __all__ = [
     "Artifacts",
     "AsyncAgents",
     "AsyncArtifacts",
+    "AsyncBrain",
     "AsyncChat",
     "AsyncMCP",
     "AsyncModels",
     "AsyncSessions",
     "AsyncSkills",
     "AsyncTeams",
+    "Brain",
     "Chat",
     "Models",
     "Sessions",

gumloop-0.3.3/src/gumloop/resources/brain.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from typing import Any
+
+from gumloop._http import AsyncHttpClient
+from gumloop._http import HttpClient
+from gumloop.types import BrainSearchRequest
+from gumloop.types import BrainSearchResponse
+
+
+class Brain:
+    def __init__(self, client: HttpClient) -> None:
+        self._client = client
+
+    def search(
+        self,
+        query: str,
+        *,
+        limit: int | None = None,
+        source_type: list[str] | None = None,
+        **kwargs: Any,
+    ) -> BrainSearchResponse:
+        return BrainSearchResponse.model_validate(
+            self._client.post(
+                "brain/search",
+                json=BrainSearchRequest.build(query=query, limit=limit, source_type=source_type, **kwargs),
+            )
+        )
+
+
+class AsyncBrain:
+    def __init__(self, client: AsyncHttpClient) -> None:
+        self._client = client
+
+    async def search(
+        self,
+        query: str,
+        *,
+        limit: int | None = None,
+        source_type: list[str] | None = None,
+        **kwargs: Any,
+    ) -> BrainSearchResponse:
+        data = await self._client.post(
+            "brain/search",
+            json=BrainSearchRequest.build(query=query, limit=limit, source_type=source_type, **kwargs),
+        )
+        return BrainSearchResponse.model_validate(data)

{gumloop-0.3.2 → gumloop-0.3.3}/src/gumloop/types.py

@@ -236,6 +236,35 @@ class SkillDownloadResponse(_Model):
     major_version: int | None = None
 
 
+# ---------------------------------------------------------------------------
+# Brain types
+# ---------------------------------------------------------------------------
+
+
+class BrainSearchRequest(_Model):
+    query: str
+    limit: int | None = None
+    source_type: list[str] | None = Field(default=None, min_length=1)
+
+
+class BrainSearchResult(_Model):
+    document_id: str | None = None
+    source: str | None = None
+    title: str | None = None
+    content: str | None = None
+    url: str | None = None
+    score: float | None = None
+    updated_at: str | None = None
+    owner_name: str | None = None
+    owner_email: str | None = None
+    parent_title: str | None = None
+    metadata: dict[str, Any] = Field(default_factory=dict)
+
+
+class BrainSearchResponse(_Model):
+    results: list[BrainSearchResult] = Field(default_factory=list)
+
+
 # ---------------------------------------------------------------------------
 # Artifact types
 # ---------------------------------------------------------------------------

gumloop-0.3.3/tests/cli/test_brain.py

@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+import json
+
+import httpx
+import respx
+from typer.testing import CliRunner
+
+from gumloop.cli.credentials import Credentials
+from gumloop.cli.credentials import save_credentials
+from gumloop.cli.main import app
+from tests.sdk.helpers import API_BASE
+
+_RESULT = {
+    "document_id": "notion:doc_1",
+    "source": "notion",
+    "title": "Onboarding",
+    "content": "How we onboard new teammates.",
+    "url": "https://notion.so/doc_1",
+    "score": 0.87,
+}
+
+
+@respx.mock
+def test_brain_search_forwards_query_limit_and_sources(cli_runner: CliRunner) -> None:
+    route = respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": []}))
+    save_credentials(Credentials(api_key="key"))
+
+    result = cli_runner.invoke(
+        app,
+        ["brain", "search", "onboarding", "--limit", "5", "--source", "notion", "--source", "slack", "--json"],
+    )
+
+    assert result.exit_code == 0, result.output
+    body = json.loads(route.calls[0].request.content)
+    assert body == {"query": "onboarding", "limit": 5, "source_type": ["notion", "slack"]}
+
+
+@respx.mock
+def test_brain_search_renders_result_table(cli_runner: CliRunner) -> None:
+    respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": [_RESULT]}))
+    save_credentials(Credentials(api_key="key"))
+
+    result = cli_runner.invoke(app, ["brain", "search", "onboarding"])
+
+    assert result.exit_code == 0, result.output
+    assert "Onboarding" in result.output
+    assert "notion" in result.output
+    assert "0.870" in result.output
+
+
+@respx.mock
+def test_brain_search_prints_empty_state(cli_runner: CliRunner) -> None:
+    respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": []}))
+    save_credentials(Credentials(api_key="key"))
+
+    result = cli_runner.invoke(app, ["brain", "search", "nothing"])
+
+    assert result.exit_code == 0, result.output
+    assert "No results found." in result.output
+
+
+@respx.mock
+def test_brain_search_surfaces_api_errors(cli_runner: CliRunner) -> None:
+    respx.post(f"{API_BASE}/brain/search").mock(
+        return_value=httpx.Response(
+            402,
+            json={
+                "error": {
+                    "code": "credit_limit_exceeded",
+                    "message": "Credit limit exceeded.",
+                    "type": "payment_required_error",
+                }
+            },
+        )
+    )
+    save_credentials(Credentials(api_key="key"))
+
+    result = cli_runner.invoke(app, ["brain", "search", "q", "--json"])
+
+    assert result.exit_code != 0

gumloop-0.3.3/tests/sdk/test_brain.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+import asyncio
+
+import httpx
+import pytest
+import respx
+from pydantic import ValidationError
+
+from gumloop import AsyncGumloop
+from gumloop import Gumloop
+from tests.sdk.helpers import API_BASE
+from tests.sdk.helpers import request_json
+
+_RESULT = {
+    "document_id": "notion:doc_1",
+    "source": "notion",
+    "title": "Onboarding",
+    "content": "How we onboard new teammates.",
+    "url": "https://notion.so/doc_1",
+    "score": 0.87,
+    "updated_at": "2026-01-02T03:04:05+00:00",
+    "owner_name": "Ada",
+}
+
+
+@respx.mock
+def test_brain_search_posts_query_and_drops_unset_fields(client: Gumloop) -> None:
+    route = respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": [_RESULT]}))
+
+    result = client.brain.search("onboarding")
+
+    assert request_json(route.calls[0].request) == {"query": "onboarding"}
+    assert len(result.results) == 1
+    assert result.results[0].title == "Onboarding"
+    assert result.results[0].source == "notion"
+    assert result.results[0].score == 0.87
+
+
+@respx.mock
+def test_brain_search_forwards_limit_and_source_type(client: Gumloop) -> None:
+    route = respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": []}))
+
+    result = client.brain.search("pricing", limit=5, source_type=["notion", "slack"])
+
+    assert request_json(route.calls[0].request) == {
+        "query": "pricing",
+        "limit": 5,
+        "source_type": ["notion", "slack"],
+    }
+    assert result.results == []
+
+
+def test_brain_search_rejects_empty_source_type(client: Gumloop) -> None:
+    with pytest.raises(ValidationError):
+        client.brain.search("pricing", source_type=[])
+
+
+@respx.mock
+def test_brain_search_passes_through_unknown_kwargs(client: Gumloop) -> None:
+    route = respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": []}))
+
+    client.brain.search("q", future_param="x")
+
+    assert request_json(route.calls[0].request) == {"query": "q", "future_param": "x"}
+
+
+@respx.mock
+def test_async_brain_search_mirrors_sync(async_client: AsyncGumloop) -> None:
+    route = respx.post(f"{API_BASE}/brain/search").mock(return_value=httpx.Response(200, json={"results": [_RESULT]}))
+
+    result = asyncio.run(async_client.brain.search("onboarding", limit=3))
+
+    assert request_json(route.calls[0].request) == {"query": "onboarding", "limit": 3}
+    assert result.results[0].document_id == "notion:doc_1"

{gumloop-0.3.2 → gumloop-0.3.3}/uv.lock

@@ -355,7 +355,7 @@ wheels = [
 
 [[package]]
 name = "gumloop"
-version = "0.3.2"
+version = "0.3.3"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },