From fb70552b577e96bb84a9449bc540e67ae8751333 Mon Sep 17 00:00:00 2001
From: Matt Rubens <mrubens@users.noreply.github.com>
Date: Sat, 28 Dec 2024 21:50:27 -0800
Subject: [PATCH] Interface cleanup

---
 src/api/providers/openrouter.ts | 7 ++-----
 src/utils/enhance-prompt.ts     | 2 +-
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/api/providers/openrouter.ts b/src/api/providers/openrouter.ts
index ccfe167..9bccf5c 100644
--- a/src/api/providers/openrouter.ts
+++ b/src/api/providers/openrouter.ts
@@ -4,7 +4,7 @@ import OpenAI from "openai"
 import { ApiHandler } from "../"
 import { ApiHandlerOptions, ModelInfo, openRouterDefaultModelId, openRouterDefaultModelInfo } from "../../shared/api"
 import { convertToOpenAiMessages } from "../transform/openai-format"
-import { ApiStream, ApiStreamChunk, ApiStreamUsageChunk } from "../transform/stream"
+import { ApiStreamChunk, ApiStreamUsageChunk } from "../transform/stream"
 import delay from "delay"
 
 // Add custom interface for OpenRouter params
@@ -17,10 +17,7 @@ interface OpenRouterApiStreamUsageChunk extends ApiStreamUsageChunk {
     fullResponseText: string;
 }
 
-// Interface for providers that support single completions
-export interface SingleCompletionHandler {
-    completePrompt(prompt: string): Promise<string>
-}
+import { SingleCompletionHandler } from ".."
 
 export class OpenRouterHandler implements ApiHandler, SingleCompletionHandler {
 	private options: ApiHandlerOptions
diff --git a/src/utils/enhance-prompt.ts b/src/utils/enhance-prompt.ts
index 3c68ced..5b1ad21 100644
--- a/src/utils/enhance-prompt.ts
+++ b/src/utils/enhance-prompt.ts
@@ -1,6 +1,6 @@
 import { ApiConfiguration } from "../shared/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.