Revert "implement deployment id"

This reverts commit b107db98fa.
This commit is contained in:
Paul Gauthier 2023-12-06 09:20:53 -08:00
parent b107db98fa
commit 57ab2cc9da
8 changed files with 13 additions and 21 deletions

View file

@ -16,13 +16,13 @@ class Model:
completion_price = None
@classmethod
def create(cls, name, client=None, deployment_id=None):
def create(cls, name, client=None):
from .openai import OpenAIModel
from .openrouter import OpenRouterModel
if client and client.base_url.host == "openrouter.ai":
return OpenRouterModel(client, name)
return OpenAIModel(name, deployment_id=deployment_id)
return OpenAIModel(name)
def __str__(self):
return self.name

View file

@ -13,9 +13,8 @@ known_tokens = {
class OpenAIModel(Model):
def __init__(self, name, deployment_id=None):
def __init__(self, name):
self.name = name
self.deployment_id = deployment_id
tokens = None