Ruff: Add ASYNC checks (#1139)

2024-01-10 05:09:42 +01:00
parent f3553040bc
commit 3cff5e9898
4 changed files with 26 additions and 50 deletions
--- a/embedchain/config/llm/base.py
+++ b/embedchain/config/llm/base.py
@@ -145,7 +145,7 @@ class BaseLlmConfig(BaseConfig):
        self.endpoint = endpoint
        self.model_kwargs = model_kwargs

-        if type(prompt) is str:
+        if isinstance(prompt, str):
            prompt = Template(prompt)

        if self.validate_prompt(prompt):