jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit 8811677154
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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