99 * - LLM_DO_GENERATE: Generates complete non-streaming responses
1010 */
1111
12- import {
13- APICallError ,
14- type LanguageModelV2CallOptions ,
15- type LanguageModelV2StreamPart ,
12+ import type {
13+ LanguageModelV2CallOptions ,
14+ LanguageModelV2StreamPart ,
1615} from "@ai-sdk/provider" ;
1716import {
1817 LANGUAGE_MODEL_BINDING ,
@@ -468,12 +467,6 @@ const getUsageFromStream = (
468467 ] ;
469468} ;
470469
471- const isAPICallError = ( error : unknown ) : error is APICallError =>
472- typeof error === "object" &&
473- error !== null &&
474- Symbol . for ( "vercel.ai.error" ) in error &&
475- Symbol . for ( "vercel.ai.error.AI_APICallError" ) in error ;
476-
477470/**
478471 * LLM_DO_STREAM - Streams a language model response in real-time
479472 */
@@ -495,27 +488,16 @@ export const createLLMStreamTool = (env: Env) =>
495488 const apiKey = getOpenRouterApiKey ( env ) ;
496489 // Create OpenRouter provider
497490 const openrouter = createOpenRouter ( { apiKey } ) ;
498- const model = openrouter ( modelId ) ;
491+ const model = openrouter . languageModel ( modelId ) ;
499492
500- try {
501- const callResponse = await model . doStream (
502- callOptions as LanguageModelV2CallOptions ,
503- ) ;
493+ const callResponse = await model . doStream (
494+ callOptions as LanguageModelV2CallOptions ,
495+ ) ;
496+ const [ _ , stream ] = getUsageFromStream ( callResponse . stream ) ;
497+ const response = streamToResponse ( stream ) ;
504498
505- const [ _ , stream ] = getUsageFromStream ( callResponse . stream ) ;
506- const response = streamToResponse ( stream ) ;
507-
508- // Return the data stream response
509- return response ;
510- } catch ( error ) {
511- if ( isAPICallError ( error ) ) {
512- return new Response ( error . responseBody , {
513- status : error . statusCode ,
514- headers : error . responseHeaders ,
515- } ) ;
516- }
517- return new Response ( String ( error ?? "Unknown error" ) , { status : 500 } ) ;
518- }
499+ // Return the data stream response
500+ return response ;
519501 } ,
520502 } ) ;
521503
0 commit comments