msy127 commited on
Commit
06abb5e
·
1 Parent(s): 4339c07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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 CustomInferenceClient(LLM, KwArgsModel, metaclass=type(LLM)):
 
 
 
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