estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: remove langchain from output parsers (#3473)

Browse Source
This commit is contained in:
takatost
2024-04-15 00:23:42 +08:00
committed by GitHub
parent 12f1ce4794
commit 8811677154
7 changed files with 8 additions and 127 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/llm_generator/llm_generator.py
View File

@@ -1,8 +1,7 @@
import json
import logging
from langchain.schema import OutputParserException
from core.llm_generator.output_parser.errors import OutputParserException
from core.llm_generator.output_parser.rule_config_generator import RuleConfigGeneratorOutputParser
from core.llm_generator.output_parser.suggested_questions_after_answer import SuggestedQuestionsAfterAnswerOutputParser
from core.llm_generator.prompts import CONVERSATION_TITLE_PROMPT, GENERATOR_QA_PROMPT

2
api/core/llm_generator/output_parser/errors.py Normal file
View File

@@ -0,0 +1,2 @@
class OutputParserException(Exception):
pass

5
api/core/llm_generator/output_parser/rule_config_generator.py
View File

@@ -1,12 +1,11 @@
from typing import Any
from langchain.schema import BaseOutputParser, OutputParserException
from core.llm_generator.output_parser.errors import OutputParserException
from core.llm_generator.prompts import RULE_CONFIG_GENERATE_TEMPLATE
from libs.json_in_md_parser import parse_and_check_json_markdown
class RuleConfigGeneratorOutputParser(BaseOutputParser):
class RuleConfigGeneratorOutputParser:
def get_format_instructions(self) -> str:
return RULE_CONFIG_GENERATE_TEMPLATE

4
api/core/llm_generator/output_parser/suggested_questions_after_answer.py
View File

@@ -2,12 +2,10 @@ import json
import re
from typing import Any
from langchain.schema import BaseOutputParser
from core.llm_generator.prompts import SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT
class SuggestedQuestionsAfterAnswerOutputParser(BaseOutputParser):
class SuggestedQuestionsAfterAnswerOutputParser:
def get_format_instructions(self) -> str:
return SUGGESTED_QUESTIONS_AFTER_ANSWER_INSTRUCTION_PROMPT