Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,10 @@ retriever = vectordb.as_retriever(search_kwargs={"k": 5})
|
|
29 |
class KwArgsModel(BaseModel):
|
30 |
kwargs: Dict[str, Any] = Field(default_factory=dict)
|
31 |
|
32 |
-
class
|
|
|
|
|
|
|
33 |
model_name: str
|
34 |
inference_client: InferenceClient
|
35 |
|
|
|
29 |
class KwArgsModel(BaseModel):
|
30 |
kwargs: Dict[str, Any] = Field(default_factory=dict)
|
31 |
|
32 |
+
class CombinedMeta(type(LLM), type(KwArgsModel)):
|
33 |
+
pass
|
34 |
+
|
35 |
+
class CustomInferenceClient(LLM, KwArgsModel, metaclass=CombinedMeta):
|
36 |
model_name: str
|
37 |
inference_client: InferenceClient
|
38 |
|