pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 20:31:37 -05:00
Code Issues Packages Projects Releases Wiki Activity

Interface cleanup

Browse Source
This commit is contained in:
Matt Rubens
2024-12-28 21:50:27 -08:00
parent f33281c18d
commit fb70552b57
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/api/providers/openrouter.ts
View File

@@ -4,7 +4,7 @@ import OpenAI from "openai"
import { ApiHandler } from "../" import { ApiHandler } from "../"
import { ApiHandlerOptions, ModelInfo, openRouterDefaultModelId, openRouterDefaultModelInfo } from "../../shared/api" import { ApiHandlerOptions, ModelInfo, openRouterDefaultModelId, openRouterDefaultModelInfo } from "../../shared/api"
import { convertToOpenAiMessages } from "../transform/openai-format" import { convertToOpenAiMessages } from "../transform/openai-format"
import { ApiStream, ApiStreamChunk, ApiStreamUsageChunk } from "../transform/stream" import { ApiStreamChunk, ApiStreamUsageChunk } from "../transform/stream"
import delay from "delay" import delay from "delay"
// Add custom interface for OpenRouter params // Add custom interface for OpenRouter params
@@ -17,10 +17,7 @@ interface OpenRouterApiStreamUsageChunk extends ApiStreamUsageChunk {
fullResponseText: string; fullResponseText: string;
} }
// Interface for providers that support single completions import { SingleCompletionHandler } from ".."
export interface SingleCompletionHandler {
completePrompt(prompt: string): Promise<string>
}
export class OpenRouterHandler implements ApiHandler, SingleCompletionHandler { export class OpenRouterHandler implements ApiHandler, SingleCompletionHandler {
private options: ApiHandlerOptions private options: ApiHandlerOptions

2
src/utils/enhance-prompt.ts
View File

@@ -1,6 +1,6 @@
import { ApiConfiguration } from "../shared/api" import { ApiConfiguration } from "../shared/api"
import { buildApiHandler } from "../api" import { buildApiHandler } from "../api"
import { OpenRouterHandler, SingleCompletionHandler } from "../api/providers/openrouter" import { OpenRouterHandler } from "../api/providers/openrouter"
/** /**
* Enhances a prompt using the OpenRouter API without creating a full Cline instance or task history. * Enhances a prompt using the OpenRouter API without creating a full Cline instance or task history.