Add provider gateway and collaboration records

AgentCompany-New/src/main/services/chat-service.ts

@@ -14,6 +14,7 @@ export class ChatService {
     const body = input.bodyMd.trim();
     if (!body) throw new Error("Message body is required.");
     const id = createId("msg");
+    const threadId = input.threadId ?? createId("thread");
     const now = new Date().toISOString();
     this.db.prepare(`
       insert into chat_messages(
@@ -23,7 +24,7 @@ export class ChatService {
       id,
       input.orgId,
       input.spaceId,
-      input.threadId,
+      threadId,
       input.senderType,
       input.senderId,
       body,
@@ -31,6 +32,25 @@ export class ChatService {
       null,
       null,
     );
+    if (!input.threadId) {
+      this.db.prepare(`
+        insert into chat_threads(id, org_id, space_id, root_message_id, title, state, created_at, updated_at)
+        values (?, ?, ?, ?, ?, ?, ?, ?)
+      `).run(threadId, input.orgId, input.spaceId, id, createThreadTitle(body), "open", now, now);
+    } else {
+      this.db.prepare("update chat_threads set updated_at = ? where id = ? and org_id = ?").run(now, input.threadId, input.orgId);
+    }
+    this.recordMentions(id, body);
+    this.recordSlashInvocation(id, body, now);
+    this.recordAttachments(input.orgId, id, input.attachmentFileIds ?? []);
+    if (input.senderId) {
+      this.db.prepare(`
+        insert into chat_presence(principal_type, principal_id, status, status_text, last_seen_at)
+        values (?, ?, ?, ?, ?)
+        on conflict(principal_type, principal_id)
+        do update set status = excluded.status, status_text = excluded.status_text, last_seen_at = excluded.last_seen_at
+      `).run(input.senderType, input.senderId, "active", "", now);
+    }
     this.audit.record({
       orgId: input.orgId,
       entityType: "chat_message",
@@ -38,8 +58,62 @@ export class ChatService {
       actorType: input.senderType,
       actorId: input.senderId,
       action: "sent",
-      payload: { spaceId: input.spaceId, threadId: input.threadId },
+      payload: { spaceId: input.spaceId, threadId, attachmentCount: input.attachmentFileIds?.length ?? 0 },
     });
     return id;
   }
+
+  private recordMentions(messageId: string, body: string): void {
+    const seen = new Set<string>();
+    for (const match of body.matchAll(/(^|\s)@([A-Za-z][A-Za-z0-9_-]{1,63})/g)) {
+      const target = match[2];
+      const offsetStart = match.index + match[1].length;
+      const key = `${target}:${offsetStart}`;
+      if (seen.has(key)) continue;
+      seen.add(key);
+      this.db.prepare(`
+        insert into chat_mentions(message_id, target_type, target_id, offset_start, offset_end)
+        values (?, ?, ?, ?, ?)
+      `).run(messageId, "alias", target, offsetStart, offsetStart + target.length + 1);
+    }
+  }
+
+  private recordSlashInvocation(messageId: string, body: string, now: string): void {
+    const match = body.match(/^\/([a-z][a-z0-9_-]*)(?:\s+([\s\S]*))?$/i);
+    if (!match) return;
+    const command = match[1].toLowerCase();
+    if (!ALLOWED_SLASH_COMMANDS.has(command)) throw new Error(`Slash command /${command} is unavailable.`);
+    this.db.prepare(`
+      insert into chat_slash_invocations(id, message_id, command, args_json, result_ref, created_at)
+      values (?, ?, ?, ?, ?, ?)
+    `).run(createId("slash"), messageId, command, JSON.stringify({ raw: match[2] ?? "" }), null, now);
+  }
+
+  private recordAttachments(orgId: string, messageId: string, fileIds: string[]): void {
+    for (const fileId of fileIds) {
+      const file = this.db.prepare("select sha256 from file_objects where id = ? and org_id = ?")
+        .get(fileId, orgId) as { sha256: string } | undefined;
+      if (!file) throw new Error("Attachment file was not found.");
+      this.db.prepare(`
+        insert into chat_attachments(message_id, file_id, transfer_type, checksum, preview_key)
+        values (?, ?, ?, ?, ?)
+      `).run(messageId, fileId, "handoff", file.sha256, null);
+    }
+  }
+}
+
+const ALLOWED_SLASH_COMMANDS = new Set([
+  "agent",
+  "handoff",
+  "approve",
+  "deny",
+  "skill",
+  "runbook",
+  "file",
+  "escalate",
+]);
+
+function createThreadTitle(body: string): string {
+  const firstLine = body.split(/\r?\n/, 1)[0].trim();
+  return firstLine.length > 80 ? `${firstLine.slice(0, 77)}...` : firstLine;
 }

AgentCompany-New/src/main/services/provider-adapters.ts

@@ -0,0 +1,119 @@
+export interface ProviderExecutionInput {
+  apiKey: string;
+  model: string;
+  prompt: string;
+  instructions: string | null;
+}
+
+export interface ProviderExecutionResult {
+  text: string;
+  rawResponseJson: string;
+}
+
+export interface ProviderAdapter {
+  createResponse(input: ProviderExecutionInput): Promise<ProviderExecutionResult>;
+}
+
+type JsonRecord = Record<string, unknown>;
+
+async function postJson(endpoint: string, headers: Record<string, string>, body: JsonRecord): Promise<JsonRecord> {
+  const response = await fetch(endpoint, {
+    method: "POST",
+    headers: {
+      "content-type": "application/json",
+      ...headers,
+    },
+    body: JSON.stringify(body),
+  });
+  const data = await response.json() as JsonRecord;
+  if (!response.ok) {
+    const message = readProviderError(data) ?? `Provider request failed with HTTP ${response.status}.`;
+    throw new Error(message);
+  }
+  return data;
+}
+
+function readProviderError(data: JsonRecord): string | null {
+  const error = data.error;
+  if (typeof error === "string") return error;
+  if (isRecord(error) && typeof error.message === "string") return error.message;
+  return null;
+}
+
+function isRecord(value: unknown): value is JsonRecord {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+
+function collectText(value: unknown): string[] {
+  if (typeof value === "string") return [value];
+  if (Array.isArray(value)) return value.flatMap((item) => collectText(item));
+  if (!isRecord(value)) return [];
+  const direct = ["text", "output_text", "content"].flatMap((key) => collectText(value[key]));
+  return direct.length > 0 ? direct : Object.values(value).flatMap((item) => collectText(item));
+}
+
+function requireText(parts: string[], providerKey: string): string {
+  const text = parts.map((part) => part.trim()).filter(Boolean).join("\n");
+  if (!text) throw new Error(`${providerKey} returned no text content.`);
+  return text;
+}
+
+export const providerAdapters: Record<string, ProviderAdapter> = {
+  openai: {
+    async createResponse(input) {
+      const data = await postJson("https://api.openai.com/v1/responses", {
+        authorization: `Bearer ${input.apiKey}`,
+      }, {
+        model: input.model,
+        input: input.prompt,
+        ...(input.instructions ? { instructions: input.instructions } : {}),
+      });
+      const text = requireText(collectText(data.output_text ?? data.output), "OpenAI");
+      return { text, rawResponseJson: JSON.stringify(data) };
+    },
+  },
+  anthropic: {
+    async createResponse(input) {
+      const data = await postJson("https://api.anthropic.com/v1/messages", {
+        "x-api-key": input.apiKey,
+        "anthropic-version": "2023-06-01",
+      }, {
+        model: input.model,
+        max_tokens: 4096,
+        ...(input.instructions ? { system: input.instructions } : {}),
+        messages: [{ role: "user", content: input.prompt }],
+      });
+      const text = requireText(collectText(data.content), "Anthropic");
+      return { text, rawResponseJson: JSON.stringify(data) };
+    },
+  },
+  xai: {
+    async createResponse(input) {
+      return createOpenAiCompatibleResponse("https://api.x.ai/v1/chat/completions", "xAI", input);
+    },
+  },
+  groq: {
+    async createResponse(input) {
+      return createOpenAiCompatibleResponse("https://api.groq.com/openai/v1/chat/completions", "Groq", input);
+    },
+  },
+};
+
+async function createOpenAiCompatibleResponse(
+  endpoint: string,
+  providerLabel: string,
+  input: ProviderExecutionInput,
+): Promise<ProviderExecutionResult> {
+  const messages = [
+    ...(input.instructions ? [{ role: "system", content: input.instructions }] : []),
+    { role: "user", content: input.prompt },
+  ];
+  const data = await postJson(endpoint, {
+    authorization: `Bearer ${input.apiKey}`,
+  }, {
+    model: input.model,
+    messages,
+  });
+  const text = requireText(collectText(data.choices), providerLabel);
+  return { text, rawResponseJson: JSON.stringify(data) };
+}

AgentCompany-New/src/main/services/provider-service.ts

@@ -1,8 +1,11 @@
 import type { DatabaseSync } from "node:sqlite";
-import type { SaveProviderCredentialInput } from "../../shared/commands";
+import { createHash } from "node:crypto";
+import type { CreateProviderResponseInput, SaveProviderCredentialInput } from "../../shared/commands";
+import { PROVIDER_AUTH_TYPE, PROVIDER_REQUEST_STATE } from "../../shared/domain";
 import { createId } from "../../shared/ids";
 import { AuditService } from "./audit-service";
 import { SecretVault } from "./secret-vault";
+import { providerAdapters } from "./provider-adapters";
 
 function displayHint(secret: string): string {
   const clean = secret.trim();
@@ -50,4 +53,94 @@ export class ProviderService {
       payload: { providerKey: provider.provider_key, authType: provider.auth_type },
     });
   }
+
+  async createResponse(input: CreateProviderResponseInput): Promise<void> {
+    const prompt = input.prompt.trim();
+    if (!prompt) throw new Error("Prompt is required.");
+    const provider = this.db.prepare(`
+      select id, provider_key, auth_type, model, credential_id, enabled
+      from providers
+      where id = ? and org_id = ?
+    `).get(input.providerId, input.orgId) as ProviderRow | undefined;
+    if (!provider) throw new Error("Provider was not found.");
+    if (provider.enabled !== 1) throw new Error("Provider credential is required.");
+    if (provider.auth_type !== PROVIDER_AUTH_TYPE.API_KEY) {
+      throw new Error("Provider response execution requires an API key adapter.");
+    }
+    if (!provider.credential_id) throw new Error("Provider credential is required.");
+    if (!provider.model) throw new Error("Provider model is required.");
+    const adapter = providerAdapters[provider.provider_key];
+    if (!adapter) throw new Error(`Provider adapter is unavailable for ${provider.provider_key}.`);
+
+    const credential = this.db.prepare(`
+      select ciphertext
+      from credentials
+      where id = ? and org_id = ? and provider_key = ?
+    `).get(provider.credential_id, input.orgId, provider.provider_key) as { ciphertext: Uint8Array } | undefined;
+    if (!credential) throw new Error("Provider credential was not found.");
+
+    const requestId = createId("preq");
+    const now = new Date().toISOString();
+    this.db.prepare(`
+      insert into provider_requests(
+        id, org_id, provider_id, provider_key, model, prompt_sha256,
+        response_text, raw_response_json, state, error_message, created_at, completed_at
+      )
+      values (?, ?, ?, ?, ?, ?, null, null, ?, null, ?, null)
+    `).run(
+      requestId,
+      input.orgId,
+      provider.id,
+      provider.provider_key,
+      provider.model,
+      hashPrompt(prompt),
+      PROVIDER_REQUEST_STATE.PENDING,
+      now,
+    );
+
+    try {
+      const apiKey = this.vault.decryptSecret(Buffer.from(credential.ciphertext));
+      const result = await adapter.createResponse({
+        apiKey,
+        model: provider.model,
+        prompt,
+        instructions: input.instructions?.trim() || null,
+      });
+      this.db.prepare(`
+        update provider_requests
+        set state = ?, response_text = ?, raw_response_json = ?, completed_at = ?
+        where id = ? and org_id = ?
+      `).run(PROVIDER_REQUEST_STATE.SUCCEEDED, result.text, result.rawResponseJson, new Date().toISOString(), requestId, input.orgId);
+      this.audit.record({
+        orgId: input.orgId,
+        entityType: "provider_request",
+        entityId: requestId,
+        actorType: "system",
+        actorId: null,
+        action: "response_created",
+        payload: { providerKey: provider.provider_key, model: provider.model },
+      });
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      this.db.prepare(`
+        update provider_requests
+        set state = ?, error_message = ?, completed_at = ?
+        where id = ? and org_id = ?
+      `).run(PROVIDER_REQUEST_STATE.FAILED, message, new Date().toISOString(), requestId, input.orgId);
+      throw error;
+    }
+  }
+}
+
+interface ProviderRow {
+  id: string;
+  provider_key: string;
+  auth_type: string;
+  model: string | null;
+  credential_id: string | null;
+  enabled: number;
+}
+
+function hashPrompt(prompt: string): string {
+  return createHash("sha256").update(prompt).digest("hex");
 }

AgentCompany-New/src/main/services/snapshot-service.ts

@@ -44,13 +44,19 @@ export class SnapshotService {
       approvals: list(this.db, "approvals", "created_at desc"),
       chatSpaces: list(this.db, "chat_spaces", "created_at asc"),
       chatMessages: list(this.db, "chat_messages", "created_at asc"),
+      chatThreads: list(this.db, "chat_threads", "updated_at desc"),
+      chatAttachments: list(this.db, "chat_attachments", "message_id asc"),
+      chatPresence: list(this.db, "chat_presence", "last_seen_at desc"),
       files: list(this.db, "file_objects", "created_at desc"),
       providers,
       providerCatalog: PROVIDER_CATALOG,
+      providerRequests: list(this.db, "provider_requests", "created_at desc"),
       skillDefinitions: list(this.db, "skill_definitions", "created_at asc"),
       skillRevisions: list(this.db, "skill_revisions", "published_at desc"),
       skillBindings,
+      skillRuntimeSnapshots: list(this.db, "skill_runtime_snapshots", "created_at desc"),
       socialAccounts: list(this.db, "social_accounts", "created_at asc"),
+      policyRules: list(this.db, "policy_rules", "created_at asc"),
       auditEvents: list<AppSnapshot["auditEvents"][number]>(this.db, "audit_events", "created_at desc").slice(0, 200),
     };
   }