Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 17, 2024
1 parent ed29f6b commit 68f2dd9
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions DocSum/docsum.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
LLM_SERVICE_HOST_IP = os.getenv("LLM_SERVICE_HOST_IP", "0.0.0.0")
LLM_SERVICE_PORT = int(os.getenv("LLM_SERVICE_PORT", 9000))


def align_inputs(self, inputs, cur_node, runtime_graph, llm_parameters_dict, **kwargs):
if self.services[cur_node].service_type == ServiceType.LLM:
docsum_parameters = kwargs.get("docsum_parameters", None)
Expand All @@ -34,6 +35,7 @@ def align_inputs(self, inputs, cur_node, runtime_graph, llm_parameters_dict, **k
inputs.update(docsum_parameters.dict())
return inputs


def read_pdf(file):
from langchain.document_loaders import PyPDFLoader

Expand Down

0 comments on commit 68f2dd9

Please sign in to comment.