Skip to content

Commit 788fc9f

Browse files
committed
make model mandatory
1 parent 40ebbcc commit 788fc9f

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

nbs/llm/llm.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
"T = t.TypeVar('T', bound=BaseModel)\n",
4343
"\n",
4444
"class RagasLLM:\n",
45-
" def __init__(self, provider: str, client: t.Any, model: t.Optional[str] = None):\n",
45+
" def __init__(self, provider: str, model:str, client: t.Any):\n",
4646
" self.provider = provider.lower()\n",
4747
" self.model = model\n",
4848
" self.client = self._initialize_client(provider, client)\n",
@@ -139,7 +139,7 @@
139139
" **kwargs\n",
140140
" )\n",
141141
"\n",
142-
"def ragas_llm(provider: str, client: t.Any, model: t.Optional[str] = None) -> RagasLLM:\n",
142+
"def ragas_llm(provider: str,model:str, client: t.Any,) -> RagasLLM:\n",
143143
" return RagasLLM(provider=provider, client=client, model=model)"
144144
]
145145
},

ragas_annotator/llm/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
T = t.TypeVar('T', bound=BaseModel)
1414

1515
class RagasLLM:
16-
def __init__(self, provider: str, client: t.Any, model: t.Optional[str] = None):
16+
def __init__(self, provider: str, model:str, client: t.Any):
1717
self.provider = provider.lower()
1818
self.model = model
1919
self.client = self._initialize_client(provider, client)
@@ -110,5 +110,5 @@ async def agenerate(self, prompt: str, response_model: t.Type[T], **kwargs) -> T
110110
**kwargs
111111
)
112112

113-
def ragas_llm(provider: str, client: t.Any, model: t.Optional[str] = None) -> RagasLLM:
113+
def ragas_llm(provider: str,model:str, client: t.Any,) -> RagasLLM:
114114
return RagasLLM(provider=provider, client=client, model=model)

0 commit comments

Comments
 (0)