mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-16 23:13:31 +00:00
community[minor]: Enable retrieval api calls in PebbloRetrievalQA (#21958)
Description: Enable app discovery and Prompt/Response apis in PebbloSafeRetrieval Documentation: NA Unit test: N/A --------- Signed-off-by: Rahul Tripathi <rauhl.psit.ec@gmail.com> Co-authored-by: Rahul Tripathi <rauhl.psit.ec@gmail.com>
This commit is contained in:
@@ -60,3 +60,86 @@ class ChainInput(BaseModel):
|
||||
base_dict["auth_context"] = self.auth_context
|
||||
base_dict["semantic_context"] = self.semantic_context
|
||||
return base_dict
|
||||
|
||||
|
||||
class Runtime(BaseModel):
|
||||
"""
|
||||
OS, language details
|
||||
"""
|
||||
|
||||
type: Optional[str] = ""
|
||||
host: str
|
||||
path: str
|
||||
ip: Optional[str] = ""
|
||||
platform: str
|
||||
os: str
|
||||
os_version: str
|
||||
language: str
|
||||
language_version: str
|
||||
runtime: Optional[str] = ""
|
||||
|
||||
|
||||
class Framework(BaseModel):
|
||||
"""
|
||||
Langchain framework details
|
||||
"""
|
||||
|
||||
name: str
|
||||
version: str
|
||||
|
||||
|
||||
class Model(BaseModel):
|
||||
vendor: Optional[str]
|
||||
name: Optional[str]
|
||||
|
||||
|
||||
class PkgInfo(BaseModel):
|
||||
project_home_page: Optional[str]
|
||||
documentation_url: Optional[str]
|
||||
pypi_url: Optional[str]
|
||||
liscence_type: Optional[str]
|
||||
installed_via: Optional[str]
|
||||
location: Optional[str]
|
||||
|
||||
|
||||
class VectorDB(BaseModel):
|
||||
name: Optional[str] = None
|
||||
version: Optional[str] = None
|
||||
location: Optional[str] = None
|
||||
embedding_model: Optional[str] = None
|
||||
|
||||
|
||||
class Chains(BaseModel):
|
||||
name: str
|
||||
model: Optional[Model]
|
||||
vector_dbs: Optional[List[VectorDB]]
|
||||
|
||||
|
||||
class App(BaseModel):
|
||||
name: str
|
||||
owner: str
|
||||
description: Optional[str]
|
||||
runtime: Runtime
|
||||
framework: Framework
|
||||
chains: List[Chains]
|
||||
plugin_version: str
|
||||
|
||||
|
||||
class Context(BaseModel):
|
||||
retrieved_from: Optional[str]
|
||||
doc: Optional[str]
|
||||
vector_db: str
|
||||
|
||||
|
||||
class Prompt(BaseModel):
|
||||
data: str
|
||||
|
||||
|
||||
class Qa(BaseModel):
|
||||
name: str
|
||||
context: List[Optional[Context]]
|
||||
prompt: Prompt
|
||||
response: Prompt
|
||||
prompt_time: str
|
||||
user: str
|
||||
user_identities: Optional[List[str]]
|
||||
|
Reference in New Issue
Block a user