Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[AUR-361] Setup pre-commit, pytest, GitHub actions, ssh-secret (#3)

Browse Source 
Co-authored-by: trducng <trungduc1992@gmail.com>
This commit is contained in:
ian_Cin
2023-08-30 07:22:01 +07:00
committed by GitHub
parent c3c25db48c
commit 5241edbc46
19 changed files with 268 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
knowledgehub/llms/completions/base.py
View File

@@ -1,7 +1,7 @@
from typing import Type
from typing import List, Type
from theflow.base import Param
from langchain.schema.language_model import BaseLanguageModel
from theflow.base import Param
from ...components import BaseComponent
from ..base import LLMInterface
@@ -41,7 +41,7 @@ class LangchainLLM(LLM):
logits=[],
)
def run_batch_raw(self, text: list[str]) -> list[LLMInterface]:
def run_batch_raw(self, text: List[str]) -> List[LLMInterface]:
outputs = []
for each_text in text:
outputs.append(self.run_raw(each_text))
@@ -50,7 +50,7 @@ class LangchainLLM(LLM):
def run_document(self, text: str) -> LLMInterface:
return self.run_raw(text)
def run_batch_document(self, text: list[str]) -> list[LLMInterface]:
def run_batch_document(self, text: List[str]) -> List[LLMInterface]:
return self.run_batch_raw(text)
def is_document(self, text) -> bool: