refactor(constants): add role

Author: remarkablemark

src/components/Chat.tsx

@@ -2,8 +2,11 @@ import { Spinner, TextInput } from '@inkjs/ui';
 import { Box, Text } from 'ink';
 import { useCallback, useState } from 'react';
 
+import { ROLE } from '../constants';
 import { ollama } from '../utils';
 
+const PROMPT_PREFIX = '> ';
+
 export function Chat() {
   const [messages, setMessages] = useState<ollama.Message[]>([]);
   const [submitKey, setSubmitKey] = useState(0);
@@ -18,14 +21,14 @@ export function Chat() {
       setIsLoading(true);
 
       const userMessage: ollama.Message = {
-        role: 'user',
+        role: ROLE.USER,
         content: userContent,
       };
       setMessages((prev) => [...prev, userMessage]);
 
       const updatedMessages = [...messages, userMessage];
       const assistantMessage: ollama.Message = {
-        role: 'assistant',
+        role: ROLE.ASSISTANT,
         content: '',
       };
       setMessages((prev) => [...prev, assistantMessage]);
@@ -57,8 +60,11 @@ export function Chat() {
     <Box flexDirection="column">
       <Box flexDirection="column">
         {messages.map((message, index) => (
-          <Text key={index} color={message.role === 'user' ? 'green' : 'blue'}>
-            {message.role === 'user' ? '> ' : ''}
+          <Text
+            key={index}
+            color={message.role === ROLE.USER ? 'green' : 'blue'}
+          >
+            {message.role === ROLE.USER ? PROMPT_PREFIX : ''}
             {message.content}
           </Text>
         ))}
@@ -69,7 +75,7 @@ export function Chat() {
       </Box>
 
       <Box>
-        <Text>&gt; </Text>
+        <Text>{PROMPT_PREFIX}</Text>
         <TextInput
           key={submitKey}
           defaultValue=""

src/constants/index.ts

@@ -0,0 +1 @@
+export * from './role';

src/constants/role.ts

@@ -0,0 +1,7 @@
+export const ROLE = {
+  USER: 'user',
+  ASSISTANT: 'assistant',
+  SYSTEM: 'system',
+} as const;
+
+export type Role = (typeof ROLE)[keyof typeof ROLE];

src/utils/ollama.ts

@@ -1,12 +1,14 @@
 import { Ollama } from 'ollama';
 
+import type { Role } from '../constants';
+
 const OLLAMA_HOST = process.env.OLLAMA_HOST ?? 'http://localhost:11434';
 const DEFAULT_MODEL = process.env.OLLAMA_MODEL ?? 'gemma4';
 
 const client = new Ollama({ host: OLLAMA_HOST });
 
 export interface Message {
-  role: 'user' | 'assistant' | 'system';
+  role: Role;
   content: string;
 }
 
@@ -29,5 +31,5 @@ export async function* streamChat(
 
 export async function listModels(): Promise<string[]> {
   const response = await client.list();
-  return response.models.map((m) => m.name);
+  return response.models.map(({ name }) => name);
 }