mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-18 08:03:36 +00:00
community: Refactor PebbloSafeLoader (#25582)
**Refactor PebbloSafeLoader** - Created `APIWrapper` and moved API logic into it. - Moved helper functions to the utility file. - Created smaller functions and methods for better readability. - Properly read environment variables. - Removed unused code. **Issue:** NA **Dependencies:** NA **tests**: Updated
This commit is contained in:
@@ -144,4 +144,5 @@ def test_pebblo_safe_loader_api_key() -> None:
|
||||
)
|
||||
|
||||
# Assert
|
||||
assert loader.api_key == api_key
|
||||
assert loader.pb_client.api_key == api_key
|
||||
assert loader.pb_client.classifier_location == "local"
|
||||
|
Reference in New Issue
Block a user