Commit
•
cae3fdf
1
Parent(s):
89be04c
Remove demonstrations (for now).
Browse files- OpenAIChatAtomicFlow.py +1 -2
OpenAIChatAtomicFlow.py
CHANGED
|
@@ -11,7 +11,6 @@ import langchain
|
|
| 11 |
from langchain.schema import HumanMessage, AIMessage, SystemMessage
|
| 12 |
|
| 13 |
from flows.base_flows.abstract import AtomicFlow
|
| 14 |
-
from flows.datasets import GenericDemonstrationsDataset
|
| 15 |
|
| 16 |
from flows import utils
|
| 17 |
from flows.messages.flow_message import UpdateMessage_ChatMessage
|
|
@@ -36,7 +35,7 @@ class OpenAIChatAtomicFlow(AtomicFlow):
|
|
| 36 |
human_message_prompt_template: PromptTemplate
|
| 37 |
|
| 38 |
query_message_prompt_template: Optional[PromptTemplate] = None
|
| 39 |
-
demonstrations: GenericDemonstrationsDataset = None
|
| 40 |
demonstrations_response_template: PromptTemplate = None
|
| 41 |
|
| 42 |
def __init__(self, **kwargs):
|
|
|
|
| 11 |
from langchain.schema import HumanMessage, AIMessage, SystemMessage
|
| 12 |
|
| 13 |
from flows.base_flows.abstract import AtomicFlow
|
|
|
|
| 14 |
|
| 15 |
from flows import utils
|
| 16 |
from flows.messages.flow_message import UpdateMessage_ChatMessage
|
|
|
|
| 35 |
human_message_prompt_template: PromptTemplate
|
| 36 |
|
| 37 |
query_message_prompt_template: Optional[PromptTemplate] = None
|
| 38 |
+
# demonstrations: GenericDemonstrationsDataset = None
|
| 39 |
demonstrations_response_template: PromptTemplate = None
|
| 40 |
|
| 41 |
def __init__(self, **kwargs):
|