devmemory-cli 0.1.0.dev0__tar.gz → 0.1.0.dev1__tar.gz

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: devmemory-cli
-Version: 0.1.0.dev0
+Version: 0.1.0.dev1
 Summary: Development-memory and version-intelligence for AI-assisted software development
 Project-URL: Documentation, https://github.com/KYadavPR/devmemory
 Project-URL: Source, https://github.com/KYadavPR/devmemory

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/__about__.py

@@ -1,3 +1,3 @@
 """Single source of truth for the package version."""
 
-__version__ = "0.1.0.dev0"
+__version__ = "0.1.0.dev1"

devmemory_cli-0.1.0.dev1/src/devmemory/adapters/genie.py

@@ -0,0 +1,161 @@
+"""Databricks Genie - a natural-language chat layer over the published telemetry.
+
+Genie answers questions in plain English against a *Genie space* that sits on
+top of the ``devmemory.analytics`` Delta tables (see
+:mod:`devmemory.adapters.databricks`). This adapter is a thin wrapper over the
+Databricks SDK's Genie Conversation API: one call per turn, returning the
+answer text, the SQL Genie ran, and a small preview of the rows.
+
+Everything degrades gracefully (§21): with no credentials, no ``databricks``
+extra, or no space id, :attr:`GenieAdapter.is_configured` is ``False`` and the
+dashboard simply hides the chat.
+"""
+
+from __future__ import annotations
+
+import os
+from typing import TYPE_CHECKING, Any
+
+from pydantic import BaseModel
+
+from devmemory.config import DevMemoryConfig, resolve_databricks_credentials
+from devmemory.domain.errors import DatabricksError
+from devmemory.logging import get_logger
+
+if TYPE_CHECKING:
+    from databricks.sdk import WorkspaceClient
+
+_log = get_logger(__name__)
+
+_MAX_ROWS = 50
+
+
+class GenieUnavailableError(DatabricksError):
+    """Genie is not configured, the SDK is missing, or the space is unreachable."""
+
+
+class GenieAnswer(BaseModel):
+    conversation_id: str | None = None
+    message_id: str | None = None
+    question: str
+    text: str | None = None
+    sql: str | None = None
+    sql_description: str | None = None
+    columns: list[str] = []
+    rows: list[list[Any]] = []
+    row_count: int | None = None
+    truncated: bool = False
+    error: str | None = None
+
+
+class GenieAdapter:
+    def __init__(self, config: DevMemoryConfig) -> None:
+        self._config = config
+        self._space_id = (
+            os.environ.get("DATABRICKS_GENIE_SPACE_ID") or config.databricks.genie_space_id or ""
+        ).strip()
+        self._client: WorkspaceClient | None = None
+
+    @property
+    def space_id(self) -> str:
+        return self._space_id
+
+    @property
+    def is_configured(self) -> bool:
+        return bool(self._space_id) and resolve_databricks_credentials() is not None
+
+    def unavailable_reason(self) -> str | None:
+        if not self._space_id:
+            return "no Genie space configured (set databricks.genie_space_id or DATABRICKS_GENIE_SPACE_ID)"
+        if resolve_databricks_credentials() is None:
+            return "Databricks credentials are not set (DATABRICKS_HOST / DATABRICKS_TOKEN)"
+        return None
+
+    # -- connection ---------------------------------------------------
+
+    def _workspace(self) -> WorkspaceClient:
+        if self._client is not None:
+            return self._client
+        creds = resolve_databricks_credentials()
+        if creds is None:
+            raise GenieUnavailableError("Databricks credentials are not set.")
+        try:
+            from databricks.sdk import WorkspaceClient
+        except ImportError as exc:  # pragma: no cover - extra not installed
+            raise GenieUnavailableError(
+                "the databricks extra is not installed",
+                hint="pip install 'devmemory-cli[databricks]'",
+            ) from exc
+        self._client = WorkspaceClient(host=creds.host, token=creds.token)
+        return self._client
+
+    # -- one chat turn ----------------------------------------------
+
+    def ask(self, question: str, *, conversation_id: str | None = None) -> GenieAnswer:
+        """Send one question and wait for the answer. Pass ``conversation_id`` to
+        continue an existing thread (Genie keeps the earlier context)."""
+        if not self._space_id:
+            raise GenieUnavailableError("no Genie space configured")
+        w = self._workspace()
+        q = question.strip()
+        try:
+            if conversation_id:
+                msg = w.genie.create_message_and_wait(self._space_id, conversation_id, q)
+            else:
+                msg = w.genie.start_conversation_and_wait(self._space_id, q)
+        except Exception as exc:
+            raise GenieUnavailableError(f"Genie request failed: {exc}") from exc
+
+        answer = GenieAnswer(
+            question=q,
+            conversation_id=msg.conversation_id,
+            message_id=msg.message_id,
+            error=msg.error.error if msg.error is not None else None,
+        )
+        self._fill_from_attachments(w, msg, answer)
+        _log.info(
+            "genie.answer",
+            space=self._space_id,
+            conversation=answer.conversation_id,
+            has_sql=answer.sql is not None,
+            rows=len(answer.rows),
+        )
+        return answer
+
+    def _fill_from_attachments(
+        self, w: WorkspaceClient, msg: Any, answer: GenieAnswer
+    ) -> None:
+        for att in msg.attachments or []:
+            if getattr(att, "text", None) and att.text.content:
+                answer.text = (answer.text or "") + att.text.content
+            query = getattr(att, "query", None)
+            if query is None:
+                continue
+            answer.sql = query.query
+            answer.sql_description = query.description
+            if not answer.text and query.description:
+                answer.text = query.description
+            try:
+                self._attach_rows(w, msg, att.attachment_id, answer)
+            except Exception as exc:  # results are best-effort
+                _log.warning("genie.results_failed", error=str(exc))
+
+    def _attach_rows(
+        self, w: WorkspaceClient, msg: Any, attachment_id: str, answer: GenieAnswer
+    ) -> None:
+        res = w.genie.get_message_query_result_by_attachment(
+            self._space_id, msg.conversation_id, msg.message_id, attachment_id
+        )
+        sr = res.statement_response
+        if sr is None:
+            return
+        schema = sr.manifest.schema if sr.manifest else None
+        cols = (schema.columns or []) if schema else []
+        answer.columns = [c.name for c in cols if c.name]
+        data = (sr.result.data_array if sr.result else None) or []
+        answer.row_count = len(data)
+        answer.truncated = len(data) > _MAX_ROWS
+        answer.rows = [list(r) for r in data[:_MAX_ROWS]]
+
+
+__all__ = ["GenieAdapter", "GenieAnswer", "GenieUnavailableError"]

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/api/app.py

@@ -17,12 +17,15 @@ from fastapi.responses import FileResponse, PlainTextResponse
 from fastapi.staticfiles import StaticFiles
 
 from devmemory.__about__ import __version__
+from devmemory.adapters.genie import GenieAdapter, GenieAnswer, GenieUnavailableError
 from devmemory.adapters.graph import GraphImpact
 from devmemory.api import mappers
 from devmemory.api.schemas import (
     AgentCheckRequest,
     ComparisonResponse,
     FeatureDetail,
+    GenieAskRequest,
+    GenieStatus,
     IssueRequest,
     ProjectBriefDoc,
     ProjectBriefRequest,
@@ -153,6 +156,30 @@ def create_app(repo_path: Path | str | None = None, *, enable_restore: bool = Fa
         doc = briefsvc.set_brief(ctx, body.content)
         return ProjectBriefDoc(content=doc.content, updated_at=doc.updated_at)
 
+    # -- Genie chat (Databricks) -------------------------------------
+
+    @app.get("/api/genie/status", response_model=GenieStatus)
+    def genie_status(ctx: Ctx) -> GenieStatus:
+        g = GenieAdapter(ctx.config)
+        return GenieStatus(
+            configured=g.is_configured,
+            space_id=g.space_id or None,
+            reason=g.unavailable_reason(),
+        )
+
+    @app.post("/api/genie/ask", response_model=GenieAnswer)
+    def genie_ask(ctx: Ctx, body: GenieAskRequest) -> GenieAnswer:
+        question = body.question.strip()
+        if not question:
+            raise HTTPException(status_code=422, detail="question is empty")
+        g = GenieAdapter(ctx.config)
+        if not g.is_configured:
+            raise HTTPException(status_code=503, detail=g.unavailable_reason() or "Genie unavailable")
+        try:
+            return g.ask(question, conversation_id=body.conversation_id)
+        except GenieUnavailableError as exc:
+            raise HTTPException(status_code=502, detail=exc.message) from exc
+
     # -- versions ------------------------------------------------------
 
     @app.get("/api/versions", response_model=list[VersionListItem])

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/api/schemas.py

@@ -181,11 +181,26 @@ class ProjectBriefRequest(BaseModel):
     content: str
 
 
+class GenieAskRequest(BaseModel):
+    """One turn in the dashboard's Genie chat."""
+
+    question: str
+    conversation_id: str | None = None
+
+
+class GenieStatus(BaseModel):
+    configured: bool
+    space_id: str | None = None
+    reason: str | None = None
+
+
 __all__ = [
     "AgentCheckRequest",
     "ComparisonResponse",
     "FeatureDetail",
     "FeatureHistoryPoint",
+    "GenieAskRequest",
+    "GenieStatus",
     "IssueRequest",
     "MetricChange",
     "ProjectBriefDoc",

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/config.py

@@ -101,6 +101,11 @@ class DatabricksSettings(_Section):
     enabled: bool = False
     catalog: str = "devmemory"
     schema_name: str = Field(default="analytics", alias="schema")
+    genie_space_id: str | None = Field(
+        default=None,
+        description="Genie space that backs the dashboard chat. "
+        "Overridden by DATABRICKS_GENIE_SPACE_ID.",
+    )
 
 
 class RegressionSettings(_Section):

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/web/frontend/src/App.tsx

@@ -5,6 +5,7 @@ import { TopBar } from "@/components/TopBar";
 import { CommandPalette } from "@/components/CommandPalette";
 import { ErrorBoundary } from "@/components/ErrorBoundary";
 import { Overview } from "@/routes/Overview";
+import { Ask } from "@/routes/Ask";
 import { Tasks } from "@/routes/Tasks";
 import { TaskDetail } from "@/routes/TaskDetail";
 import { Timeline } from "@/routes/Timeline";
@@ -54,6 +55,7 @@ export function App() {
             <div className="route-fade" key={location.pathname}>
               <Routes>
                 <Route path="/" element={<Overview />} />
+                <Route path="/ask" element={<Ask />} />
                 <Route path="/tasks" element={<Tasks />} />
                 <Route path="/task/:id" element={<TaskDetail />} />
                 <Route path="/timeline" element={<Timeline />} />

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/web/frontend/src/api/client.ts

@@ -16,6 +16,8 @@ import type {
   TaskSummary,
   SnapshotSummary,
   ProjectBriefDoc,
+  GenieStatus,
+  GenieAnswer,
 } from "./types";
 
 const BASE = "/api";
@@ -157,6 +159,22 @@ export const useSearch = (q: string) =>
 export const checkChange = (files: string[], intent: string, feature?: string) =>
   apiPost<ChangeGuidance>("/agent/check", { files, intent: intent || null, feature: feature || null });
 
+// --- Genie chat (Databricks) ----------------------------------------------
+
+export const useGenieStatus = (opts?: QOpts<GenieStatus>) =>
+  useQuery({
+    queryKey: ["genie-status"],
+    queryFn: () => apiGet<GenieStatus>("/genie/status"),
+    staleTime: 60_000,
+    ...opts,
+  });
+
+export const askGenie = (question: string, conversationId: string | null) =>
+  apiPost<GenieAnswer>("/genie/ask", {
+    question,
+    conversation_id: conversationId,
+  });
+
 // --- project brief (single source of truth) --------------------------------
 
 export const useProjectBrief = (opts?: QOpts<ProjectBriefDoc>) =>

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/web/frontend/src/api/types.ts

@@ -520,3 +520,23 @@ export interface ProjectBriefDoc {
   content: string;
   updated_at: string | null;
 }
+
+export interface GenieStatus {
+  configured: boolean;
+  space_id: string | null;
+  reason: string | null;
+}
+
+export interface GenieAnswer {
+  conversation_id: string | null;
+  message_id: string | null;
+  question: string;
+  text: string | null;
+  sql: string | null;
+  sql_description: string | null;
+  columns: string[];
+  rows: unknown[][];
+  row_count: number | null;
+  truncated: boolean;
+  error: string | null;
+}

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/web/frontend/src/components/Sidebar.tsx

@@ -5,6 +5,7 @@ import { Logo } from "./Logo";
 
 const NAV: { to: string; label: string; icon: IconName }[] = [
   { to: "/", label: "Overview", icon: "overview" },
+  { to: "/ask", label: "Ask", icon: "sparkles" },
   { to: "/tasks", label: "Tasks", icon: "target" },
   { to: "/timeline", label: "Timeline", icon: "timeline" },
   { to: "/features", label: "Features", icon: "features" },

devmemory_cli-0.1.0.dev1/src/devmemory/web/frontend/src/routes/Ask.tsx

@@ -0,0 +1,235 @@
+import { useEffect, useRef, useState } from "react";
+import { useMutation } from "@tanstack/react-query";
+import { useGenieStatus, askGenie } from "@/api/client";
+import { Card, PageHead, EmptyState } from "@/components/primitives";
+import { Icon } from "@/components/Icon";
+import type { GenieAnswer } from "@/api/types";
+
+type Turn =
+  | { role: "user"; text: string }
+  | { role: "genie"; answer: GenieAnswer }
+  | { role: "error"; text: string };
+
+const SUGGESTIONS = [
+  "Which features regressed the most?",
+  "Show every version that failed tests, newest first",
+  "What files change together most often?",
+  "How did quote_latency_ms move over time?",
+];
+
+function errorDetail(message: string): string {
+  try {
+    const parsed = JSON.parse(message);
+    if (parsed && typeof parsed.detail === "string") return parsed.detail;
+  } catch {
+    /* not json */
+  }
+  return message;
+}
+
+export function Ask() {
+  const status = useGenieStatus();
+  const [turns, setTurns] = useState<Turn[]>([]);
+  const [draft, setDraft] = useState("");
+  const [conversationId, setConversationId] = useState<string | null>(null);
+  const scrollerRef = useRef<HTMLDivElement>(null);
+
+  const ask = useMutation({
+    mutationFn: (q: string) => askGenie(q, conversationId),
+    onSuccess: (answer) => {
+      if (answer.conversation_id) setConversationId(answer.conversation_id);
+      setTurns((t) => [...t, { role: "genie", answer }]);
+    },
+    onError: (err: Error) => {
+      setTurns((t) => [...t, { role: "error", text: errorDetail(err.message) }]);
+    },
+  });
+
+  useEffect(() => {
+    scrollerRef.current?.scrollTo({ top: scrollerRef.current.scrollHeight, behavior: "smooth" });
+  }, [turns, ask.isPending]);
+
+  const send = (q: string) => {
+    const question = q.trim();
+    if (!question || ask.isPending) return;
+    setTurns((t) => [...t, { role: "user", text: question }]);
+    setDraft("");
+    ask.mutate(question);
+  };
+
+  const newChat = () => {
+    setTurns([]);
+    setConversationId(null);
+  };
+
+  return (
+    <>
+      <PageHead
+        title="Ask"
+        subtitle="Natural-language questions over your development history, answered by Databricks Genie against the devmemory.analytics tables."
+        actions={
+          turns.length > 0 ? (
+            <button className="btn" onClick={newChat}>
+              <Icon name="plus" size={14} /> New chat
+            </button>
+          ) : undefined
+        }
+      />
+
+      {status.data && !status.data.configured ? (
+        <Card title="Genie is not configured" pad>
+          <p className="text-sm">{status.data.reason}</p>
+          <p className="muted text-sm">
+            Set a Genie space that reads the <code>devmemory.analytics</code> schema, then point
+            DevMemory at it:
+          </p>
+          <pre className="loop-prompt">{`# .devmemory/config.json
+"databricks": { "enabled": true, "genie_space_id": "01ef..." }
+
+# or the environment
+DATABRICKS_HOST=...       DATABRICKS_TOKEN=...
+DATABRICKS_GENIE_SPACE_ID=01ef...`}</pre>
+          <p className="muted text-xs">
+            Run <code>devmemory databricks sync</code> first so the tables have data for Genie to
+            query.
+          </p>
+        </Card>
+      ) : (
+        <Card flush>
+          <div className="chat" ref={scrollerRef}>
+            {turns.length === 0 && (
+              <div className="chat__empty">
+                <EmptyState
+                  icon="sparkles"
+                  title="Ask about your project"
+                  sub="Genie writes the SQL, runs it on Databricks, and explains the result."
+                />
+                <div className="chat__suggest">
+                  {SUGGESTIONS.map((s) => (
+                    <button key={s} className="chip" onClick={() => send(s)}>
+                      {s}
+                    </button>
+                  ))}
+                </div>
+              </div>
+            )}
+
+            {turns.map((turn, i) => (
+              <TurnView key={i} turn={turn} />
+            ))}
+
+            {ask.isPending && (
+              <div className="chat__msg chat__msg--genie">
+                <div className="chat__role">Genie</div>
+                <div className="chat__body row" style={{ gap: 8 }}>
+                  <span className="spinner" /> thinking…
+                </div>
+              </div>
+            )}
+          </div>
+
+          <form
+            className="chat__input"
+            onSubmit={(e) => {
+              e.preventDefault();
+              send(draft);
+            }}
+          >
+            <input
+              className="input"
+              placeholder="Ask a question about this project…"
+              value={draft}
+              onChange={(e) => setDraft(e.target.value)}
+              disabled={ask.isPending || !status.data?.configured}
+            />
+            <button
+              className="btn btn--primary"
+              type="submit"
+              disabled={ask.isPending || !draft.trim() || !status.data?.configured}
+            >
+              <Icon name="arrowRight" size={15} />
+            </button>
+          </form>
+        </Card>
+      )}
+    </>
+  );
+}
+
+function TurnView({ turn }: { turn: Turn }) {
+  if (turn.role === "user") {
+    return (
+      <div className="chat__msg chat__msg--user">
+        <div className="chat__body">{turn.text}</div>
+      </div>
+    );
+  }
+  if (turn.role === "error") {
+    return (
+      <div className="chat__msg chat__msg--genie">
+        <div className="chat__role">Genie</div>
+        <div className="chat__body callout callout--bad" style={{ padding: "8px 10px" }}>
+          <Icon name="alert" size={13} /> {turn.text}
+        </div>
+      </div>
+    );
+  }
+
+  const a = turn.answer;
+  return (
+    <div className="chat__msg chat__msg--genie">
+      <div className="chat__role">Genie</div>
+      <div className="chat__body stack" style={{ gap: 10 }}>
+        {a.error && (
+          <div className="callout callout--bad" style={{ padding: "8px 10px" }}>
+            <Icon name="alert" size={13} /> {a.error}
+          </div>
+        )}
+        {a.text && <div style={{ whiteSpace: "pre-wrap" }}>{a.text}</div>}
+        {a.rows.length > 0 && <ResultTable columns={a.columns} rows={a.rows} />}
+        {(a.row_count != null || a.truncated) && (
+          <div className="muted text-xs">
+            {a.row_count ?? a.rows.length} row{(a.row_count ?? a.rows.length) === 1 ? "" : "s"}
+            {a.truncated && ` · showing first ${a.rows.length}`}
+          </div>
+        )}
+        {a.sql && (
+          <details className="chat__sql">
+            <summary>SQL</summary>
+            <pre className="loop-prompt">{a.sql}</pre>
+          </details>
+        )}
+        {!a.text && !a.rows.length && !a.sql && !a.error && (
+          <div className="muted">No answer returned.</div>
+        )}
+      </div>
+    </div>
+  );
+}
+
+function ResultTable({ columns, rows }: { columns: string[]; rows: unknown[][] }) {
+  return (
+    <div className="dm-table__scroll">
+      <table className="dm-table">
+        <thead>
+          <tr>
+            {columns.map((c) => (
+              <th key={c}>{c}</th>
+            ))}
+          </tr>
+        </thead>
+        <tbody>
+          {rows.map((r, i) => (
+            <tr key={i}>
+              {r.map((cell, j) => (
+                <td key={j} className="mono text-xs">
+                  {cell == null ? <span className="muted">null</span> : String(cell)}
+                </td>
+              ))}
+            </tr>
+          ))}
+        </tbody>
+      </table>
+    </div>
+  );
+}

{devmemory_cli-0.1.0.dev0 → devmemory_cli-0.1.0.dev1}/src/devmemory/web/frontend/src/theme/components.css

@@ -965,3 +965,93 @@ textarea.input {
   max-height: 320px;
   overflow-y: auto;
 }
+
+/* Genie chat ("Ask") */
+.chat {
+  display: flex;
+  flex-direction: column;
+  gap: 14px;
+  padding: 18px;
+  min-height: 340px;
+  max-height: calc(100vh - 300px);
+  overflow-y: auto;
+}
+.chat__empty {
+  margin: auto;
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  gap: 16px;
+}
+.chat__suggest {
+  display: flex;
+  flex-wrap: wrap;
+  gap: 8px;
+  justify-content: center;
+  max-width: 560px;
+}
+.chip {
+  border: 1px solid var(--border);
+  background: var(--surface);
+  color: var(--text);
+  border-radius: 999px;
+  padding: 6px 12px;
+  font-size: 0.8125rem;
+  cursor: pointer;
+}
+.chip:hover {
+  border-color: var(--accent);
+  color: var(--accent);
+}
+.chat__msg {
+  display: flex;
+  flex-direction: column;
+  gap: 4px;
+  max-width: 82%;
+}
+.chat__msg--user {
+  align-self: flex-end;
+  align-items: flex-end;
+}
+.chat__msg--genie {
+  align-self: flex-start;
+  max-width: 100%;
+  width: 100%;
+}
+.chat__role {
+  font-size: 0.6875rem;
+  text-transform: uppercase;
+  letter-spacing: 0.06em;
+  color: var(--text-muted);
+}
+.chat__msg--user .chat__body {
+  background: var(--accent);
+  color: #fff;
+  border-radius: var(--r-2);
+  padding: 8px 12px;
+  white-space: pre-wrap;
+}
+.chat__msg--genie .chat__body {
+  background: var(--surface-2, var(--surface));
+  border: 1px solid var(--border);
+  border-radius: var(--r-2);
+  padding: 12px 14px;
+  font-size: 0.9375rem;
+}
+.chat__sql summary {
+  cursor: pointer;
+  font-size: 0.8125rem;
+  color: var(--text-muted);
+}
+.chat__sql[open] summary {
+  margin-bottom: 8px;
+}
+.chat__input {
+  display: flex;
+  gap: 8px;
+  padding: 14px 18px;
+  border-top: 1px solid var(--border);
+}
+.chat__input .input {
+  flex: 1;
+}