feat: AI: Ask for API key inline in chat input by ilyabo · Pull Request #357 · sqlrooms/sqlrooms

ilyabo · 2026-02-11T13:31:34Z

Kapture.2026-02-11.at.14.33.08.mp4

Summary by CodeRabbit

New Features
- Inline API key input in the chat composer to enter and save provider API keys at runtime.
- Per-provider API key error detection and tracking to surface authentication problems.
- Dynamic retrieval of API keys and base URLs from settings so changes apply without reload.

Signed-off-by: Ilya Boyandin <[email protected]>

coderabbitai · 2026-02-11T13:31:57Z

📝 Walkthrough

Walkthrough

Adds an inline API-key input component to Chat.Composer, wires it to update provider settings, and introduces per-provider API key error tracking with runtime authentication-error detection in the chat transport.

Changes

Cohort / File(s)	Summary
New Component `packages/ai-core/src/components/InlineApiKeyInput.tsx`	Adds `InlineApiKeyInput` and `InlineApiKeyInputButton`—a password input + save action that calls an onSaveApiKey(provider, apiKey) callback.
Chat Component Surface `packages/ai-core/src/components/Chat.tsx`	Exports `InlineApiKeyInput` on the Chat public surface; removes a duplicate ModelSelector import.
Query Controls `packages/ai-core/src/components/QueryControls.tsx`	Detects and extracts `InlineApiKeyInput` from children, conditionally renders it when API key input is required or errored, and adapts focus/keyboard handling accordingly.
AI State / Slice `packages/ai-core/src/AiSlice.ts`	Adds `apiKeyErrors: Record<string, boolean>` to state and public methods `setApiKeyError(provider, hasError)` and `hasApiKeyError()` to set/query provider-level API key errors; removed passing static apiKey/baseUrl into transport factory.
Types `packages/ai-core/src/types.ts`	Adds `setApiKeyError(provider: string, hasError: boolean)` to `AiStateForTransport` interface.
Chat Transport `packages/ai-core/src/chatTransport.ts`	Stops using static apiKey/baseUrl in config; fetches apiKey/baseUrl dynamically via state.ai.getApiKeyFromSettings()/getBaseUrlFromSettings(), adds isAuthenticationError detection (401/403 + common messages), and sets provider-level apiKey error via `setApiKeyError` on auth failure.
Example Integrations `examples/*/src/components/MainView.tsx` `examples/ai-agent/src/...`, `examples/ai-core/src/...`, `examples/ai-rag/src/...`, `examples/ai/src/...`	Imports `useStoreWithAiSettings` and wires an `InlineApiKeyInput` in Chat.Composer to call `updateProvider(provider, { apiKey })` on save, enabling runtime per-provider API key updates.

Sequence Diagram

sequenceDiagram
    participant User
    participant InlineApiKeyInput
    participant MainView
    participant AiSlice
    participant ChatTransport

    User->>InlineApiKeyInput: Type API key and Save
    InlineApiKeyInput->>MainView: onSaveApiKey(provider, apiKey)
    MainView->>AiSlice: updateProvider(provider, { apiKey })
    AiSlice->>AiSlice: persist provider apiKey in settings

    User->>ChatTransport: Send chat request
    ChatTransport->>AiSlice: getApiKeyFromSettings()
    AiSlice-->>ChatTransport: return apiKey
    ChatTransport->>ChatTransport: perform API request

    alt Authentication error detected
        ChatTransport->>ChatTransport: isAuthenticationError() => true
        ChatTransport->>AiSlice: setApiKeyError(provider, true)
        AiSlice->>AiSlice: mark provider apiKey error
        ChatTransport-->>User: surface auth error
    end

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~20 minutes

Possibly related PRs

fix: AI settings changes are only applied in new sessions #356: Similar refactor of chatTransport/types to stop passing static apiKey/baseUrl and retrieve them from settings at request time.

Poem

🐰 I hopped inside the chat today, a tiny key in paw,
I tucked it in the composer where the user saw,
Providers learn their secrets, errors flagged with care,
Now keys can change at runtime — sniff! — a fresh spring air. 🔑

🚥 Pre-merge checks | ✅ 2 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 42.86% which is insufficient. The required threshold is 80.00%.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (2 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title accurately describes the main feature: adding inline API key input in the chat interface for requesting user API keys during chat interactions.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing touches

📝 Generate docstrings

🧪 Generate unit tests (beta)

Create PR with unit tests
Post copyable unit tests in a comment
Commit unit tests in branch feat--Ask-for-API-key-inline-in-chat-input

No actionable comments were generated in the recent review. 🎉

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

…r-API-key-inline-in-chat-input Signed-off-by: Ilya Boyandin <[email protected]>

ilyabo added 4 commits February 11, 2026 13:08

fix: AI settings changes are only applied in new sessions

5b8c969

Signed-off-by: Ilya Boyandin <[email protected]>

feat: Ask for API key inline in chat input

1970971

Signed-off-by: Ilya Boyandin <[email protected]>

detect incorrect key

eec657f

Signed-off-by: Ilya Boyandin <[email protected]>

add to other examples

bce3175

Signed-off-by: Ilya Boyandin <[email protected]>

Merge branch 'main' of github.com:sqlrooms/sqlrooms into feat--Ask-fo…

42c94b3

…r-API-key-inline-in-chat-input Signed-off-by: Ilya Boyandin <[email protected]>

ilyabo merged commit f256021 into main Feb 11, 2026
4 checks passed

ilyabo deleted the feat--Ask-for-API-key-inline-in-chat-input branch February 11, 2026 14:04

This was referenced Feb 24, 2026

docs: LLM-friendly documentation #375

Merged

feat: Snowflake connector integration #401

Merged

coderabbitai Bot mentioned this pull request May 21, 2026

feat: Persist AI settings to TOML in the CLI #653

Merged

coderabbitai Bot mentioned this pull request Jun 9, 2026

feat: Add chat history #698

Merged

coderabbitai Bot mentioned this pull request Jun 20, 2026

feat: Compact assistant composer controls #743

Merged

Navigation Menu

Repository navigation

feat: AI: Ask for API key inline in chat input - #357

feat: AI: Ask for API key inline in chat input#357
ilyabo merged 5 commits into
mainfrom
feat--Ask-for-API-key-inline-in-chat-input

ilyabo commented Feb 11, 2026 •

edited by coderabbitai Bot

Loading

Uh oh!

coderabbitai Bot commented Feb 11, 2026 •

edited

Loading

Walkthrough

Changes

Sequence Diagram

Estimated code review effort

Possibly related PRs

Poem

Uh oh!

Uh oh!

Reviewers

Uh oh!

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

Conversation

ilyabo commented Feb 11, 2026 • edited by coderabbitai Bot Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Summary by CodeRabbit

Uh oh!

coderabbitai Bot commented Feb 11, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Walkthrough

Changes

Sequence Diagram

Estimated code review effort

Possibly related PRs

Poem

Uh oh!

Uh oh!

Reviewers

Uh oh!

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

ilyabo commented Feb 11, 2026 •

edited by coderabbitai Bot

Loading

coderabbitai Bot commented Feb 11, 2026 •

edited

Loading