github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-16 23:13:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Refactor PebbloRetrievalQA (#25583)

Browse Source 
**Refactor PebbloRetrievalQA**
  - Created `APIWrapper` and moved API logic into it.
  - Created smaller functions/methods for better readability.
  - Properly read environment variables.
  - Removed unused code.
  - Updated models

**Issue:** NA
**Dependencies:** NA
**tests**:  NA
This commit is contained in:
Rajendra Kadam
2024-08-22 21:21:21 +05:30
committed by GitHub
parent 1f1679e960
commit 4ff2f4499e
3 changed files with 391 additions and 317 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/community/langchain_community/chains/pebblo_retrieval/models.py
View File

@@ -109,7 +109,7 @@ class VectorDB(BaseModel):
embedding_model: Optional[str] = None
class Chains(BaseModel):
class ChainInfo(BaseModel):
name: str
model: Optional[Model]
vector_dbs: Optional[List[VectorDB]]
@@ -121,7 +121,7 @@ class App(BaseModel):
description: Optional[str]
runtime: Runtime
framework: Framework
chains: List[Chains]
chains: List[ChainInfo]
plugin_version: str
@@ -134,9 +134,9 @@ class Context(BaseModel):
class Prompt(BaseModel):
data: Optional[Union[list, str]]
entityCount: Optional[int]
entities: Optional[dict]
prompt_gov_enabled: Optional[bool]
entityCount: Optional[int] = None
entities: Optional[dict] = None
prompt_gov_enabled: Optional[bool] = None
class Qa(BaseModel):