prompt_objects 0.4.0 → 0.6.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (80) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +33 -0
  3. data/CLAUDE.md +113 -44
  4. data/README.md +140 -14
  5. data/frontend/index.html +5 -1
  6. data/frontend/src/App.tsx +72 -79
  7. data/frontend/src/canvas/CanvasView.tsx +5 -5
  8. data/frontend/src/canvas/constants.ts +31 -31
  9. data/frontend/src/canvas/inspector/InspectorPanel.tsx +4 -4
  10. data/frontend/src/canvas/inspector/POInspector.tsx +35 -35
  11. data/frontend/src/canvas/inspector/ToolCallInspector.tsx +13 -13
  12. data/frontend/src/canvas/nodes/PONode.ts +2 -2
  13. data/frontend/src/components/ContextMenu.tsx +5 -4
  14. data/frontend/src/components/EnvDataPane.tsx +69 -0
  15. data/frontend/src/components/Inspector.tsx +263 -0
  16. data/frontend/src/components/MarkdownMessage.tsx +22 -20
  17. data/frontend/src/components/MethodList.tsx +90 -0
  18. data/frontend/src/components/ModelSelector.tsx +13 -14
  19. data/frontend/src/components/NotificationPanel.tsx +29 -33
  20. data/frontend/src/components/ObjectList.tsx +78 -0
  21. data/frontend/src/components/PaneSlot.tsx +30 -0
  22. data/frontend/src/components/SourcePane.tsx +202 -0
  23. data/frontend/src/components/SystemBar.tsx +74 -0
  24. data/frontend/src/components/Transcript.tsx +76 -0
  25. data/frontend/src/components/UsagePanel.tsx +27 -27
  26. data/frontend/src/components/Workspace.tsx +260 -0
  27. data/frontend/src/components/index.ts +10 -9
  28. data/frontend/src/hooks/useResize.ts +55 -0
  29. data/frontend/src/hooks/useWebSocket.ts +70 -0
  30. data/frontend/src/index.css +27 -10
  31. data/frontend/src/store/index.ts +36 -0
  32. data/frontend/src/types/index.ts +13 -0
  33. data/frontend/tailwind.config.js +28 -9
  34. data/lib/prompt_objects/capability.rb +23 -1
  35. data/lib/prompt_objects/connectors/mcp.rb +2 -16
  36. data/lib/prompt_objects/environment.rb +15 -0
  37. data/lib/prompt_objects/llm/openai_adapter.rb +22 -0
  38. data/lib/prompt_objects/mcp/tools/inspect_po.rb +1 -31
  39. data/lib/prompt_objects/mcp/tools/list_prompt_objects.rb +1 -6
  40. data/lib/prompt_objects/prompt_object.rb +239 -7
  41. data/lib/prompt_objects/server/api/routes.rb +16 -48
  42. data/lib/prompt_objects/server/app.rb +14 -0
  43. data/lib/prompt_objects/server/public/assets/{index-xvyeb-5Z.js → index-DEPawnfZ.js} +206 -206
  44. data/lib/prompt_objects/server/public/assets/index-oMrRce1m.css +1 -0
  45. data/lib/prompt_objects/server/public/index.html +7 -3
  46. data/lib/prompt_objects/server/websocket_handler.rb +41 -98
  47. data/lib/prompt_objects/server.rb +6 -62
  48. data/lib/prompt_objects/session/store.rb +176 -4
  49. data/lib/prompt_objects/universal/delete_env_data.rb +70 -0
  50. data/lib/prompt_objects/universal/get_env_data.rb +64 -0
  51. data/lib/prompt_objects/universal/list_env_data.rb +61 -0
  52. data/lib/prompt_objects/universal/store_env_data.rb +87 -0
  53. data/lib/prompt_objects/universal/update_env_data.rb +88 -0
  54. data/lib/prompt_objects.rb +6 -1
  55. data/prompt_objects.gemspec +1 -1
  56. data/templates/arc-agi-1/objects/observer.md +4 -0
  57. data/templates/arc-agi-1/objects/solver.md +10 -1
  58. data/templates/arc-agi-1/objects/verifier.md +4 -0
  59. data/templates/arc-agi-1/primitives/find_objects.rb +1 -1
  60. data/templates/arc-agi-1/primitives/grid_diff.rb +2 -2
  61. data/templates/arc-agi-1/primitives/grid_info.rb +1 -1
  62. data/templates/arc-agi-1/primitives/grid_transform.rb +1 -1
  63. data/templates/arc-agi-1/primitives/render_grid.rb +1 -0
  64. data/templates/arc-agi-1/primitives/test_solution.rb +3 -0
  65. data/tools/thread-explorer.html +27 -0
  66. metadata +18 -16
  67. data/Gemfile.lock +0 -233
  68. data/IMPLEMENTATION_PLAN.md +0 -1073
  69. data/design-doc-v2.md +0 -1232
  70. data/frontend/src/components/CapabilitiesPanel.tsx +0 -141
  71. data/frontend/src/components/ChatPanel.tsx +0 -296
  72. data/frontend/src/components/Dashboard.tsx +0 -83
  73. data/frontend/src/components/Header.tsx +0 -153
  74. data/frontend/src/components/MessageBus.tsx +0 -56
  75. data/frontend/src/components/POCard.tsx +0 -56
  76. data/frontend/src/components/PODetail.tsx +0 -124
  77. data/frontend/src/components/PromptPanel.tsx +0 -156
  78. data/frontend/src/components/SessionsPanel.tsx +0 -174
  79. data/frontend/src/components/ThreadsSidebar.tsx +0 -163
  80. data/lib/prompt_objects/server/public/assets/index-6y64NXFy.css +0 -1
@@ -0,0 +1,263 @@
1
+ import { useState, useMemo } from 'react'
2
+ import { useStore, usePONotifications, useEnvData } from '../store'
3
+ import { useResize } from '../hooks/useResize'
4
+ import { MethodList } from './MethodList'
5
+ import { SourcePane } from './SourcePane'
6
+ import { EnvDataPane } from './EnvDataPane'
7
+ import { Workspace } from './Workspace'
8
+ import { ContextMenu } from './ContextMenu'
9
+ import { PaneSlot } from './PaneSlot'
10
+ import type { PromptObject, CapabilityInfo } from '../types'
11
+
12
+ interface InspectorProps {
13
+ po: PromptObject
14
+ sendMessage: (target: string, content: string, newThread?: boolean) => void
15
+ createSession?: (target: string, name?: string) => void
16
+ switchSession: (target: string, sessionId: string) => void
17
+ createThread: (target: string) => void
18
+ updatePrompt: (target: string, prompt: string) => void
19
+ requestUsage?: (sessionId: string, includeTree?: boolean) => void
20
+ exportThread?: (sessionId: string, format?: string) => void
21
+ requestEnvData: (sessionId: string) => void
22
+ }
23
+
24
+ export function Inspector({
25
+ po,
26
+ sendMessage,
27
+ switchSession,
28
+ createThread,
29
+ updatePrompt,
30
+ requestUsage,
31
+ exportThread,
32
+ requestEnvData,
33
+ }: InspectorProps) {
34
+ const [selectedCapability, setSelectedCapability] = useState<CapabilityInfo | null>(null)
35
+ const [threadMenuOpen, setThreadMenuOpen] = useState(false)
36
+ const [contextMenu, setContextMenu] = useState<{ x: number; y: number; sessionId: string } | null>(null)
37
+ const notifications = usePONotifications(po.name)
38
+ const topPaneCollapsed = useStore((s) => s.topPaneCollapsed)
39
+ const toggleTopPane = useStore((s) => s.toggleTopPane)
40
+ const envDataPaneCollapsed = useStore((s) => s.envDataPaneCollapsed)
41
+ const toggleEnvDataPane = useStore((s) => s.toggleEnvDataPane)
42
+
43
+ const topPaneResize = useResize({
44
+ direction: 'vertical',
45
+ initialSize: 260,
46
+ minSize: 120,
47
+ maxSize: 600,
48
+ })
49
+
50
+ const envDataResize = useResize({
51
+ direction: 'vertical',
52
+ initialSize: 160,
53
+ minSize: 80,
54
+ maxSize: 400,
55
+ })
56
+
57
+ const methodListResize = useResize({
58
+ direction: 'horizontal',
59
+ initialSize: 192,
60
+ minSize: 120,
61
+ maxSize: 320,
62
+ })
63
+
64
+ const sessions = po.sessions || []
65
+ const currentSessionId = po.current_session?.id
66
+ const sessionRootMap = useStore((s) => s.sessionRootMap)
67
+ const rootThreadId = currentSessionId ? sessionRootMap[currentSessionId] : undefined
68
+ const envDataEntries = useEnvData(rootThreadId)
69
+
70
+ // Sort sessions: current first, then by updated_at desc
71
+ const sortedSessions = useMemo(() => {
72
+ return [...sessions].sort((a, b) => {
73
+ if (a.id === currentSessionId) return -1
74
+ if (b.id === currentSessionId) return 1
75
+ return (b.updated_at || '').localeCompare(a.updated_at || '')
76
+ })
77
+ }, [sessions, currentSessionId])
78
+
79
+ const isActive = po.status !== 'idle'
80
+
81
+ const statusDot = {
82
+ idle: 'bg-po-status-idle',
83
+ thinking: 'bg-po-status-active',
84
+ calling_tool: 'bg-po-status-calling',
85
+ }[po.status] || 'bg-po-status-idle'
86
+
87
+ const statusGlow = {
88
+ idle: '',
89
+ thinking: 'shadow-[0_0_6px_rgba(212,149,42,0.7)]',
90
+ calling_tool: 'shadow-[0_0_6px_rgba(59,154,110,0.7)]',
91
+ }[po.status] || ''
92
+
93
+ const statusLabelColor = {
94
+ idle: 'text-po-text-ghost',
95
+ thinking: 'text-po-status-active',
96
+ calling_tool: 'text-po-status-calling',
97
+ }[po.status] || 'text-po-text-ghost'
98
+
99
+ const statusLabel = {
100
+ idle: 'idle',
101
+ thinking: 'thinking...',
102
+ calling_tool: 'calling tool...',
103
+ }[po.status] || po.status
104
+
105
+ const handleThreadContextMenu = (e: React.MouseEvent, sessionId: string) => {
106
+ e.preventDefault()
107
+ setContextMenu({ x: e.clientX, y: e.clientY, sessionId })
108
+ }
109
+
110
+ return (
111
+ <div className="h-full flex flex-col">
112
+ {/* Inspector Header */}
113
+ <div className="h-8 bg-po-surface-2 border-b border-po-border flex items-center px-3 gap-2 flex-shrink-0">
114
+ <div className="relative flex-shrink-0">
115
+ <div className={`w-2 h-2 rounded-full ${statusDot} ${statusGlow} ${isActive ? 'animate-pulse' : ''}`} />
116
+ </div>
117
+ <span className="font-mono text-xs text-po-text-primary font-medium">{po.name}</span>
118
+ <span className={`text-2xs font-medium truncate ${statusLabelColor} ${isActive ? 'animate-pulse' : ''}`}>{statusLabel}</span>
119
+ {po.description && (
120
+ <span className="text-2xs text-po-text-ghost truncate hidden sm:inline">{po.description}</span>
121
+ )}
122
+ {notifications.length > 0 && (
123
+ <span className="text-2xs font-mono bg-po-warning text-po-bg px-1 rounded font-bold">
124
+ {notifications.length}
125
+ </span>
126
+ )}
127
+
128
+ <div className="flex-1" />
129
+
130
+ {/* Thread picker */}
131
+ <div className="relative">
132
+ <button
133
+ onClick={() => setThreadMenuOpen(!threadMenuOpen)}
134
+ className="flex items-center gap-1 text-2xs text-po-text-secondary hover:text-po-text-primary transition-colors duration-150"
135
+ >
136
+ <span className="font-mono">
137
+ {currentSessionId
138
+ ? sessions.find(s => s.id === currentSessionId)?.name || `Thread ${currentSessionId.slice(0, 6)}`
139
+ : 'No thread'}
140
+ </span>
141
+ <svg className={`w-3 h-3 transition-transform ${threadMenuOpen ? 'rotate-180' : ''}`} fill="none" stroke="currentColor" viewBox="0 0 24 24">
142
+ <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M19 9l-7 7-7-7" />
143
+ </svg>
144
+ </button>
145
+
146
+ {threadMenuOpen && (
147
+ <div className="absolute right-0 top-full mt-1 w-56 bg-po-surface-2 border border-po-border rounded shadow-xl z-50 overflow-hidden">
148
+ <button
149
+ onClick={() => { createThread(po.name); setThreadMenuOpen(false) }}
150
+ className="w-full text-left px-2.5 py-1.5 text-xs text-po-accent hover:bg-po-surface-3 transition-colors duration-150 border-b border-po-border"
151
+ >
152
+ + New Thread
153
+ </button>
154
+ <div className="max-h-48 overflow-auto">
155
+ {sortedSessions.map((session) => (
156
+ <button
157
+ key={session.id}
158
+ onClick={() => { switchSession(po.name, session.id); setThreadMenuOpen(false) }}
159
+ onContextMenu={(e) => handleThreadContextMenu(e, session.id)}
160
+ className={`w-full text-left px-2.5 py-1.5 text-xs transition-colors duration-150 ${
161
+ session.id === currentSessionId
162
+ ? 'bg-po-accent-wash text-po-accent'
163
+ : 'text-po-text-secondary hover:bg-po-surface-3'
164
+ }`}
165
+ >
166
+ <div className="flex items-center gap-1.5">
167
+ {session.thread_type === 'delegation' && <span className="text-po-status-delegated">&#8627;</span>}
168
+ <span className="font-mono truncate flex-1">
169
+ {session.name || `Thread ${session.id.slice(0, 6)}`}
170
+ </span>
171
+ <span className="text-2xs text-po-text-ghost">{session.message_count}m</span>
172
+ </div>
173
+ {session.parent_po && (
174
+ <div className="text-2xs text-po-status-delegated mt-0.5">from {session.parent_po}</div>
175
+ )}
176
+ </button>
177
+ ))}
178
+ </div>
179
+ </div>
180
+ )}
181
+ </div>
182
+ </div>
183
+
184
+ {/* Top: Methods + Source (collapsible, resizable height) */}
185
+ <PaneSlot
186
+ label="Methods | Source"
187
+ collapsed={topPaneCollapsed}
188
+ onToggle={toggleTopPane}
189
+ height={topPaneResize.size}
190
+ resizeHandle={
191
+ <div
192
+ className="resize-handle-h"
193
+ onMouseDown={topPaneResize.onMouseDown}
194
+ />
195
+ }
196
+ >
197
+ <div className="flex h-full">
198
+ {/* Method List (resizable width) */}
199
+ <div style={{ width: methodListResize.size }} className="flex-shrink-0">
200
+ <MethodList
201
+ po={po}
202
+ selectedCapability={selectedCapability}
203
+ onSelectCapability={setSelectedCapability}
204
+ />
205
+ </div>
206
+
207
+ {/* Resize handle */}
208
+ <div
209
+ className="resize-handle"
210
+ onMouseDown={methodListResize.onMouseDown}
211
+ />
212
+
213
+ {/* Source Pane */}
214
+ <SourcePane
215
+ po={po}
216
+ selectedCapability={selectedCapability}
217
+ onSave={(prompt) => updatePrompt(po.name, prompt)}
218
+ />
219
+ </div>
220
+ </PaneSlot>
221
+
222
+ {/* Middle: Env Data (collapsible, resizable height) */}
223
+ <PaneSlot
224
+ label={`Env Data${envDataEntries.length > 0 ? ` (${envDataEntries.length})` : ''}`}
225
+ collapsed={envDataPaneCollapsed}
226
+ onToggle={toggleEnvDataPane}
227
+ height={envDataResize.size}
228
+ resizeHandle={
229
+ <div
230
+ className="resize-handle-h"
231
+ onMouseDown={envDataResize.onMouseDown}
232
+ />
233
+ }
234
+ >
235
+ <EnvDataPane sessionId={currentSessionId} requestEnvData={requestEnvData} />
236
+ </PaneSlot>
237
+
238
+ {/* Bottom: Workspace */}
239
+ <div className="flex-1 overflow-hidden">
240
+ <Workspace po={po} sendMessage={sendMessage} />
241
+ </div>
242
+
243
+ {/* Context menu for thread right-click */}
244
+ {contextMenu && (
245
+ <ContextMenu
246
+ x={contextMenu.x}
247
+ y={contextMenu.y}
248
+ onClose={() => setContextMenu(null)}
249
+ items={[
250
+ ...(requestUsage ? [
251
+ { label: 'View Usage', onClick: () => requestUsage(contextMenu.sessionId) },
252
+ { label: 'View Tree Usage', onClick: () => requestUsage(contextMenu.sessionId, true) },
253
+ ] : []),
254
+ ...(exportThread ? [
255
+ { label: 'Export Markdown', onClick: () => exportThread(contextMenu.sessionId, 'markdown') },
256
+ { label: 'Export JSON', onClick: () => exportThread(contextMenu.sessionId, 'json') },
257
+ ] : []),
258
+ ]}
259
+ />
260
+ )}
261
+ </div>
262
+ )
263
+ }
@@ -30,7 +30,7 @@ export function MarkdownMessage({ content, className = '' }: MarkdownMessageProp
30
30
 
31
31
  return (
32
32
  <code
33
- className="bg-po-bg px-1.5 py-0.5 rounded text-po-accent font-mono text-sm"
33
+ className="bg-po-surface-2 px-1 py-0.5 rounded text-po-accent font-mono text-[0.9em]"
34
34
  {...props}
35
35
  >
36
36
  {children}
@@ -52,29 +52,29 @@ export function MarkdownMessage({ content, className = '' }: MarkdownMessageProp
52
52
  },
53
53
  // Styled paragraphs
54
54
  p({ children }) {
55
- return <p className="mb-3 last:mb-0">{children}</p>
55
+ return <p className="mb-2 last:mb-0">{children}</p>
56
56
  },
57
57
  // Styled lists
58
58
  ul({ children }) {
59
- return <ul className="list-disc list-inside mb-3 space-y-1">{children}</ul>
59
+ return <ul className="list-disc list-inside mb-2 space-y-0.5">{children}</ul>
60
60
  },
61
61
  ol({ children }) {
62
- return <ol className="list-decimal list-inside mb-3 space-y-1">{children}</ol>
62
+ return <ol className="list-decimal list-inside mb-2 space-y-0.5">{children}</ol>
63
63
  },
64
64
  // Styled headings
65
65
  h1({ children }) {
66
- return <h1 className="text-xl font-bold mb-2 mt-4 first:mt-0">{children}</h1>
66
+ return <h1 className="text-base font-bold mb-1.5 mt-3 first:mt-0 text-po-text-primary">{children}</h1>
67
67
  },
68
68
  h2({ children }) {
69
- return <h2 className="text-lg font-bold mb-2 mt-3 first:mt-0">{children}</h2>
69
+ return <h2 className="text-sm font-bold mb-1.5 mt-2 first:mt-0 text-po-text-primary">{children}</h2>
70
70
  },
71
71
  h3({ children }) {
72
- return <h3 className="text-base font-bold mb-2 mt-2 first:mt-0">{children}</h3>
72
+ return <h3 className="text-xs font-bold mb-1 mt-1.5 first:mt-0 text-po-text-primary">{children}</h3>
73
73
  },
74
74
  // Styled blockquotes
75
75
  blockquote({ children }) {
76
76
  return (
77
- <blockquote className="border-l-4 border-po-accent pl-4 my-3 text-gray-400 italic">
77
+ <blockquote className="border-l-2 border-po-accent pl-3 my-2 text-po-text-secondary italic">
78
78
  {children}
79
79
  </blockquote>
80
80
  )
@@ -82,26 +82,26 @@ export function MarkdownMessage({ content, className = '' }: MarkdownMessageProp
82
82
  // Styled tables
83
83
  table({ children }) {
84
84
  return (
85
- <div className="overflow-x-auto my-3">
86
- <table className="min-w-full border border-po-border">{children}</table>
85
+ <div className="overflow-x-auto my-2">
86
+ <table className="min-w-full border border-po-border text-xs">{children}</table>
87
87
  </div>
88
88
  )
89
89
  },
90
90
  th({ children }) {
91
91
  return (
92
- <th className="border border-po-border bg-po-bg px-3 py-2 text-left font-medium">
92
+ <th className="border border-po-border bg-po-surface-2 px-2 py-1 text-left font-medium text-po-text-primary">
93
93
  {children}
94
94
  </th>
95
95
  )
96
96
  },
97
97
  td({ children }) {
98
98
  return (
99
- <td className="border border-po-border px-3 py-2">{children}</td>
99
+ <td className="border border-po-border px-2 py-1 text-po-text-secondary">{children}</td>
100
100
  )
101
101
  },
102
102
  // Horizontal rule
103
103
  hr() {
104
- return <hr className="border-po-border my-4" />
104
+ return <hr className="border-po-border my-3" />
105
105
  },
106
106
  }}
107
107
  >
@@ -121,17 +121,17 @@ function CodeBlock({ language, code }: { language: string; code: string }) {
121
121
  }
122
122
 
123
123
  return (
124
- <div className="relative group my-3">
124
+ <div className="relative group my-2">
125
125
  {/* Language label and copy button */}
126
- <div className="flex items-center justify-between bg-po-bg/80 px-3 py-1 rounded-t border border-b-0 border-po-border">
127
- <span className="text-xs text-gray-500 font-mono">
126
+ <div className="flex items-center justify-between bg-po-surface-2 px-2.5 py-1 rounded-t border border-b-0 border-po-border">
127
+ <span className="text-2xs text-po-text-ghost font-mono">
128
128
  {language || 'text'}
129
129
  </span>
130
130
  <button
131
131
  onClick={handleCopy}
132
- className="text-xs text-gray-400 hover:text-white transition-colors"
132
+ className="text-2xs text-po-text-ghost hover:text-po-text-primary transition-colors duration-150"
133
133
  >
134
- {copied ? 'Copied!' : 'Copy'}
134
+ {copied ? 'Copied' : 'Copy'}
135
135
  </button>
136
136
  </div>
137
137
  {/* Code with syntax highlighting */}
@@ -141,9 +141,11 @@ function CodeBlock({ language, code }: { language: string; code: string }) {
141
141
  PreTag="div"
142
142
  customStyle={{
143
143
  margin: 0,
144
- borderRadius: '0 0 0.375rem 0.375rem',
145
- border: '1px solid rgb(55, 65, 81)',
144
+ borderRadius: '0 0 0.25rem 0.25rem',
145
+ border: '1px solid #3d3a37',
146
146
  borderTop: 'none',
147
+ fontSize: '11px',
148
+ background: '#222120',
147
149
  }}
148
150
  >
149
151
  {code}
@@ -0,0 +1,90 @@
1
+ import type { PromptObject, CapabilityInfo } from '../types'
2
+
3
+ interface MethodListProps {
4
+ po: PromptObject
5
+ selectedCapability: CapabilityInfo | null
6
+ onSelectCapability: (cap: CapabilityInfo | null) => void
7
+ }
8
+
9
+ export function MethodList({ po, selectedCapability, onSelectCapability }: MethodListProps) {
10
+ const capabilities = po.capabilities || []
11
+ const universalCapabilities = po.universal_capabilities || []
12
+
13
+ const handleClick = (cap: CapabilityInfo) => {
14
+ if (selectedCapability?.name === cap.name) {
15
+ onSelectCapability(null) // Toggle off
16
+ } else {
17
+ onSelectCapability(cap)
18
+ }
19
+ }
20
+
21
+ return (
22
+ <div className="h-full border-r border-po-border overflow-auto bg-po-surface">
23
+ {/* Source (prompt view) */}
24
+ <button
25
+ onClick={() => onSelectCapability(null)}
26
+ className={`w-full text-left px-2.5 py-1 text-xs font-mono border-b border-po-border transition-colors duration-150 ${
27
+ selectedCapability === null
28
+ ? 'bg-po-accent-wash text-po-accent'
29
+ : 'text-po-text-secondary hover:bg-po-surface-2'
30
+ }`}
31
+ >
32
+ Source
33
+ </button>
34
+
35
+ {/* Declared capabilities */}
36
+ {capabilities.length > 0 && (
37
+ <div>
38
+ <div className="px-2.5 py-1.5 border-b border-po-border">
39
+ <span className="text-2xs font-medium text-po-text-ghost uppercase tracking-wider">
40
+ Methods ({capabilities.length})
41
+ </span>
42
+ </div>
43
+ {capabilities.map((cap) => (
44
+ <button
45
+ key={cap.name}
46
+ onClick={() => handleClick(cap)}
47
+ className={`w-full text-left px-2.5 py-1 text-xs font-mono transition-colors duration-150 ${
48
+ selectedCapability?.name === cap.name
49
+ ? 'bg-po-accent-wash text-po-accent'
50
+ : 'text-po-accent hover:bg-po-surface-2'
51
+ }`}
52
+ >
53
+ {cap.name}
54
+ </button>
55
+ ))}
56
+ </div>
57
+ )}
58
+
59
+ {/* Universal capabilities */}
60
+ {universalCapabilities.length > 0 && (
61
+ <div>
62
+ <div className="px-2.5 py-1.5 border-b border-po-border border-t border-po-border">
63
+ <span className="text-2xs font-medium text-po-text-ghost uppercase tracking-wider">
64
+ Universal ({universalCapabilities.length})
65
+ </span>
66
+ </div>
67
+ {universalCapabilities.map((cap) => (
68
+ <button
69
+ key={cap.name}
70
+ onClick={() => handleClick(cap)}
71
+ className={`w-full text-left px-2.5 py-1 text-xs font-mono transition-colors duration-150 ${
72
+ selectedCapability?.name === cap.name
73
+ ? 'bg-po-accent-wash text-po-text-secondary'
74
+ : 'text-po-text-secondary hover:bg-po-surface-2'
75
+ }`}
76
+ >
77
+ {cap.name}
78
+ </button>
79
+ ))}
80
+ </div>
81
+ )}
82
+
83
+ {capabilities.length === 0 && universalCapabilities.length === 0 && (
84
+ <div className="px-2.5 py-4 text-2xs text-po-text-ghost text-center">
85
+ No capabilities
86
+ </div>
87
+ )}
88
+ </div>
89
+ )
90
+ }
@@ -28,7 +28,6 @@ export function ModelSelector({ switchLLM }: Props) {
28
28
  setIsOpen(false)
29
29
  }
30
30
 
31
- // Provider display names
32
31
  const providerNames: Record<string, string> = {
33
32
  openai: 'OpenAI',
34
33
  anthropic: 'Anthropic',
@@ -41,14 +40,14 @@ export function ModelSelector({ switchLLM }: Props) {
41
40
  <div className="relative" ref={dropdownRef}>
42
41
  <button
43
42
  onClick={() => setIsOpen(!isOpen)}
44
- className="flex items-center gap-2 px-3 py-1.5 text-sm bg-po-border rounded hover:bg-po-accent/50 transition-colors"
43
+ className="flex items-center gap-1.5 px-2 py-0.5 text-xs bg-po-surface-2 border border-po-border rounded hover:border-po-border-focus transition-colors duration-150"
45
44
  >
46
- <span className="text-gray-400">
45
+ <span className="text-po-text-tertiary">
47
46
  {providerNames[llmConfig.current_provider] || llmConfig.current_provider}
48
47
  </span>
49
- <span className="text-white font-medium">{llmConfig.current_model}</span>
48
+ <span className="font-mono text-po-text-primary">{llmConfig.current_model}</span>
50
49
  <svg
51
- className={`w-4 h-4 text-gray-400 transition-transform ${isOpen ? 'rotate-180' : ''}`}
50
+ className={`w-3 h-3 text-po-text-ghost transition-transform ${isOpen ? 'rotate-180' : ''}`}
52
51
  fill="none"
53
52
  stroke="currentColor"
54
53
  viewBox="0 0 24 24"
@@ -58,13 +57,13 @@ export function ModelSelector({ switchLLM }: Props) {
58
57
  </button>
59
58
 
60
59
  {isOpen && (
61
- <div className="absolute right-0 top-full mt-2 w-64 bg-po-surface border border-po-border rounded-lg shadow-xl z-50 overflow-hidden">
60
+ <div className="absolute right-0 top-full mt-1 w-60 bg-po-surface-2 border border-po-border rounded shadow-xl z-50 overflow-hidden">
62
61
  {llmConfig.providers.map((provider) => (
63
62
  <div key={provider.name}>
64
- <div className="px-3 py-2 bg-po-bg text-xs font-medium text-gray-400 uppercase tracking-wide flex items-center justify-between">
63
+ <div className="px-2.5 py-1.5 bg-po-surface text-2xs font-medium text-po-text-ghost uppercase tracking-wider flex items-center justify-between">
65
64
  <span>{providerNames[provider.name] || provider.name}</span>
66
65
  {!provider.available && (
67
- <span className="text-red-400 text-[10px] normal-case">
66
+ <span className="text-po-error text-2xs normal-case">
68
67
  {provider.name === 'ollama' ? 'Not Running' : 'No API Key'}
69
68
  </span>
70
69
  )}
@@ -81,22 +80,22 @@ export function ModelSelector({ switchLLM }: Props) {
81
80
  key={`${provider.name}-${model}`}
82
81
  onClick={() => isAvailable && handleSelectModel(provider.name, model)}
83
82
  disabled={!isAvailable}
84
- className={`w-full px-3 py-2 text-left text-sm flex items-center justify-between transition-colors ${
83
+ className={`w-full px-2.5 py-1.5 text-left text-xs font-mono flex items-center justify-between transition-colors duration-150 ${
85
84
  isSelected
86
- ? 'bg-po-accent/20 text-po-accent'
85
+ ? 'bg-po-accent-wash text-po-accent'
87
86
  : isAvailable
88
- ? 'text-gray-300 hover:bg-po-border'
89
- : 'text-gray-600 cursor-not-allowed'
87
+ ? 'text-po-text-secondary hover:bg-po-surface-3'
88
+ : 'text-po-text-ghost cursor-not-allowed'
90
89
  }`}
91
90
  >
92
91
  <span className="flex items-center gap-2">
93
92
  {model}
94
93
  {isDefault && (
95
- <span className="text-[10px] text-gray-500">(default)</span>
94
+ <span className="text-2xs text-po-text-ghost">(default)</span>
96
95
  )}
97
96
  </span>
98
97
  {isSelected && (
99
- <svg className="w-4 h-4" fill="currentColor" viewBox="0 0 20 20">
98
+ <svg className="w-3 h-3" fill="currentColor" viewBox="0 0 20 20">
100
99
  <path
101
100
  fillRule="evenodd"
102
101
  d="M16.707 5.293a1 1 0 010 1.414l-8 8a1 1 0 01-1.414 0l-4-4a1 1 0 011.414-1.414L8 12.586l7.293-7.293a1 1 0 011.414 0z"