klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[chore] fix lint issues (#607)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-09-12 20:04:31 -07:00
committed by GitHub
parent 0f23d5f967
commit 701d0b21ef
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
embedchain/llm/base.py
View File

@@ -204,7 +204,7 @@ class BaseLlm(JSONSerializable):
finally: finally:
if config: if config:
# Restore previous config # Restore previous config
self.config: BaseLlmConfig = BaseLlmConfig.deserialize(prev_config) self.config: BaseLlmConfig = BaseLlmConfig.deserialize(prev_config)
def chat(self, input_query: str, contexts: List[str], config: BaseLlmConfig = None, dry_run=False): def chat(self, input_query: str, contexts: List[str], config: BaseLlmConfig = None, dry_run=False):
""" """

5
embedchain/loaders/mdx.py
View File

@@ -1,17 +1,14 @@
import hashlib import hashlib
from langchain.document_loaders import PyPDFLoader
from embedchain.helper.json_serializable import register_deserializable from embedchain.helper.json_serializable import register_deserializable
from embedchain.loaders.base_loader import BaseLoader from embedchain.loaders.base_loader import BaseLoader
from embedchain.utils import clean_string
@register_deserializable @register_deserializable
class MdxLoader(BaseLoader): class MdxLoader(BaseLoader):
def load_data(self, url): def load_data(self, url):
"""Load data from a mdx file.""" """Load data from a mdx file."""
with open(url, 'r', encoding="utf-8") as infile: with open(url, "r", encoding="utf-8") as infile:
content = infile.read() content = infile.read()
meta_data = { meta_data = {
"url": url, "url": url,