Skip to content

Commit 920d882

Browse files
committed
Add a special ProviderType for openrouter
We can later alias it to openai if we decide to merge them.
1 parent b180ef0 commit 920d882

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

Diff for: src/codegate/db/models.py

+1
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ class ProviderType(str, Enum):
128128
ollama = "ollama"
129129
lm_studio = "lm_studio"
130130
llamacpp = "llamacpp"
131+
openrouter = "openai"
131132

132133

133134
class GetPromptWithOutputsRow(BaseModel):

Diff for: src/codegate/muxing/adapter.py

+2
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,8 @@ def __init__(self):
106106
db_models.ProviderType.anthropic: self._format_antropic,
107107
# Our Lllamacpp provider emits OpenAI chunks
108108
db_models.ProviderType.llamacpp: self._format_openai,
109+
# OpenRouter is a dialect of OpenAI
110+
db_models.ProviderType.openrouter: self._format_openai,
109111
}
110112

111113
def _format_ollama(self, chunk: str) -> str:

Diff for: src/codegate/server.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ async def log_user_agent(request: Request, call_next):
7373

7474
# Register all known providers
7575
registry.add_provider(
76-
ProviderType.openai,
76+
ProviderType.openrouter,
7777
OpenAIProvider(pipeline_factory),
7878
)
7979
registry.add_provider(

0 commit comments

Comments
 (0)