abhijeethp's picture
added similarity score calc
0f35f83
raw
history blame contribute delete
313 Bytes
from embedding_models import open_ai
class _Registry:
def __init__(self):
pass
def _models(self):
return [
open_ai.OpenAIAda002(),
open_ai.OpenAI3Large()
]
def models(self):
return {m.name(): m for m in self._models()}
registry = _Registry()