UNPKG

rawi

Version:

Rawi (راوي) is the developer-friendly AI CLI that brings the power of 11 major AI providers directly to your terminal. With seamless shell integration, persistent conversations, and 200+ specialized prompt templates, Rawi transforms your command line into

1 lines 5.55 kB
{"version":3,"sources":["/home/mkabumattar/work/withrawi/rawi/dist/chunk-7IO5BLNF.cjs","../src/core/providers/cerebras.provider.ts"],"names":["cerebrasModelIds","cerebrasModels","name","cerebrasProvider","streamWithCerebras","credentials","prompt","settings","apiKey","baseURL","createCerebras","result","streamText","error"],"mappings":"AAAA;AACA,wDAAwC,4CCDI,wBAEL,IAoB1BA,CAAAA,CAAmB,CAC9B,aAAA,CACA,cAAA,CACA,cACF,CAAA,CAEaC,CAAAA,aAA8BD,CAAAA,CAAiB,GAAA,CAAKE,CAAAA,EAAAA,CAAU,CACzE,IAAA,CAAAA,CAAAA,CACA,WAAA,CAAaA,CACf,CAAA,CAAE,CAAA,CAEWC,CAAAA,aAAmB,CAC9B,IAAA,CAAM,UAAA,CACN,WAAA,CAAa,oBAAA,CACb,MAAA,CAAQF,CACV,CAAA,CAEaG,CAAAA,aAAqB,KAAA,CAChCC,CAAAA,CACAC,CAAAA,CAAAA,EAC+B,CAC/B,GAAI,CACF,IAAMC,CAAAA,CAAWF,CAAAA,CAAY,gBAAA,CAIvBG,CAAAA,iBAASD,CAAAA,2BAAU,QAAA,EAAUF,CAAAA,CAAY,MAAA,CAE/C,EAAA,CAAI,CAACG,CAAAA,CACH,MAAM,IAAI,KAAA,CAAM,kCAAkC,CAAA,CAGpD,IAAMC,CAAAA,iBAAUF,CAAAA,6BAAU,SAAA,EAAW,4BAAA,CAE/BJ,CAAAA,CAAmBO,sCAAAA,CACvB,MAAA,CAAQF,CAAAA,CACR,OAAA,CAASC,CACX,CAAC,CAAA,CAEKE,CAAAA,CAASC,4BAAAA,CACb,KAAA,CAAOT,CAAAA,CAAiBE,CAAAA,CAAY,KAAK,CAAA,CACzC,MAAA,CAAAC,CAAAA,CACA,WAAA,CAAaD,CAAAA,CAAY,WAAA,EAAe,EAAA,CACxC,eAAA,CAAiBA,CAAAA,CAAY,SAAA,EAAa,IAC5C,CAAC,CAAA,CAED,MAAO,CACL,UAAA,CAAYM,CAAAA,CAAO,UAAA,CACnB,YAAA,CAAcA,CAAAA,CAAO,IACvB,CACF,CAAA,KAAA,CAASE,CAAAA,CAAO,CACd,MAAM,IAAI,KAAA,CACR,CAAA,sCAAA,EACEA,EAAAA,WAAiB,KAAA,CAAQA,CAAAA,CAAM,OAAA,CAAU,MAAA,CAAOA,CAAK,CACvD,CAAA,CAAA;AD1EiuB","file":"/home/mkabumattar/work/withrawi/rawi/dist/chunk-7IO5BLNF.cjs","sourcesContent":[null,"import {type cerebras, createCerebras} from '@ai-sdk/cerebras';\nimport type {ModelMessage} from 'ai';\nimport {generateText, streamText} from 'ai';\nimport {parseCommandFromResponse} from '../exec/parser.js';\nimport type {\n CerebrasSettings,\n LooseToStrict,\n ModelInfo,\n RawiCredentials,\n StreamingResponse,\n} from '../shared/index.js';\nimport type {\n ChatCredentials,\n ChatProvider,\n ChatStreamOptions,\n ExecGenerationOptions,\n ExecGenerationResult,\n} from './types.js';\n\ntype LooseCerebrasModelId = Parameters<typeof cerebras>[0];\nexport type CerebrasModelId = LooseToStrict<LooseCerebrasModelId>;\n\nexport const cerebrasModelIds = [\n 'llama3.1-8b',\n 'llama3.1-70b',\n 'llama3.3-70b',\n] as const;\n\nexport const cerebrasModels: ModelInfo[] = cerebrasModelIds.map((name) => ({\n name,\n displayName: name,\n}));\n\nexport const cerebrasProvider = {\n name: 'cerebras' as const,\n displayName: '🧠 Cerebras',\n models: cerebrasModels,\n};\n\nexport const streamWithCerebras = async (\n credentials: RawiCredentials,\n prompt: string,\n): Promise<StreamingResponse> => {\n try {\n const settings = credentials.providerSettings as\n | CerebrasSettings\n | undefined;\n\n const apiKey = settings?.apiKey || credentials.apiKey;\n\n if (!apiKey) {\n throw new Error('API key is required for Cerebras');\n }\n\n const baseURL = settings?.baseURL || 'https://api.cerebras.ai/v1';\n\n const cerebrasProvider = createCerebras({\n apiKey: apiKey,\n baseURL: baseURL,\n });\n\n const result = streamText({\n model: cerebrasProvider(credentials.model),\n prompt,\n temperature: credentials.temperature || 0.7,\n maxOutputTokens: credentials.maxTokens || 2048,\n });\n\n return {\n textStream: result.textStream,\n fullResponse: result.text,\n };\n } catch (error) {\n throw new Error(\n `Error calling Cerebras streaming API: ${\n error instanceof Error ? error.message : String(error)\n }`,\n );\n }\n};\n\nexport const cerebrasChatProvider: ChatProvider = {\n name: 'cerebras',\n displayName: '🧠 Cerebras',\n\n async streamChat(\n credentials: ChatCredentials,\n messages: ModelMessage[],\n options: ChatStreamOptions = {},\n ): Promise<AsyncIterable<string>> {\n const settings = credentials.providerSettings || {};\n const apiKey = settings.apiKey || credentials.apiKey;\n\n if (!apiKey) {\n throw new Error('API key is required for Cerebras');\n }\n\n const cerebrasProvider = createCerebras({\n apiKey: apiKey,\n baseURL: settings.baseURL || 'https://api.cerebras.ai/v1',\n });\n\n const result = streamText({\n model: cerebrasProvider(credentials.model),\n messages,\n temperature: credentials.temperature || options.temperature || 0.7,\n maxOutputTokens: credentials.maxTokens || options.maxTokens || 2048,\n });\n\n return result.textStream;\n },\n};\n\nexport const generateWithCerebras = async (\n options: ExecGenerationOptions,\n): Promise<ExecGenerationResult> => {\n const startTime = Date.now();\n\n try {\n const settings = options.credentials.providerSettings as\n | CerebrasSettings\n | undefined;\n\n const apiKey = settings?.apiKey || options.credentials.apiKey;\n\n if (!apiKey) {\n throw new Error('API key is required for Cerebras');\n }\n\n const cerebrasProvider = createCerebras({\n apiKey: apiKey,\n baseURL: settings?.baseURL || 'https://api.cerebras.ai/v1',\n });\n\n const result = await generateText({\n model: cerebrasProvider(options.credentials.model),\n system: options.systemPrompt,\n prompt: options.userPrompt,\n });\n\n const generationTime = Date.now() - startTime;\n\n const command = parseCommandFromResponse(result.text);\n\n return {\n command,\n generationTime,\n };\n } catch (error) {\n throw new Error(`Cerebras exec generation failed: ${error}`);\n }\n};\n"]}