Skip to content

Commit 5bf7f49

Browse files
committed
Makes changes compatible to alpha changes for LCEL support
1 parent aa958f1 commit 5bf7f49

File tree

3 files changed

+7
-7
lines changed

3 files changed

+7
-7
lines changed

‎poetry.lock‎

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎pyproject.toml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ alembic = "^1.12.1"
2525
importlib-metadata = "^7.0.1"
2626
pyarrow = "^15.0.0"
2727
iniconfig = "^2.0.0"
28-
dialog-lib = "^0.0.2.0a0"
28+
dialog-lib = "0.0.2.0a1"
2929
langserve = "^0.2.0"
3030
sse-starlette = "^2.1.0"
3131

‎src/dialog/llm/agents/default.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from dialog.llm.embeddings import EMBEDDINGS_LLM
1515
from dialog.learn.idf import categorize_conversation_history
1616

17-
from dialog_lib.agents.abstract import AbstractRAG, AbstractLCELDialog
17+
from dialog_lib.agents.abstract import AbstractRAG, AbstractLCEL
1818
from dialog_lib.embeddings.generate import get_most_relevant_contents_from_message
1919
from dialog_lib.embeddings.retrievers import DialogRetriever
2020
from dialog_lib.db.memory import generate_memory_instance, CustomPostgresChatMessageHistory
@@ -104,7 +104,7 @@ def postprocess(self, output: str) -> str:
104104
return output
105105

106106

107-
class DialogLCEL(AbstractLCELDialog):
107+
class DialogLCEL(AbstractLCEL):
108108
def __init__(self, *args, **kwargs):
109109
self.openai_api_key = Settings().OPENAI_API_KEY
110110
kwargs["model_class"] = ChatOpenAI(

0 commit comments

Comments
 (0)