Spaces:
Sleeping
Sleeping
Commit
·
be06203
1
Parent(s):
6a2d3ac
feat: split message class
Browse files- app.py +3 -12
- message.py +10 -0
app.py
CHANGED
@@ -1,21 +1,12 @@
|
|
1 |
import os
|
2 |
-
from dataclasses import dataclass
|
3 |
-
from typing import Literal
|
4 |
-
import streamlit as st
|
5 |
|
|
|
|
|
6 |
from langchain import OpenAI
|
7 |
from langchain.callbacks import get_openai_callback
|
8 |
from langchain.chains import ConversationChain
|
9 |
from langchain.chains.conversation.memory import ConversationSummaryMemory
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
@dataclass
|
14 |
-
class Message:
|
15 |
-
"""Class for keeping track of a chat message."""
|
16 |
-
|
17 |
-
origin: Literal["human", "ai"]
|
18 |
-
message: str
|
19 |
|
20 |
|
21 |
def load_css():
|
|
|
1 |
import os
|
|
|
|
|
|
|
2 |
|
3 |
+
import streamlit as st
|
4 |
+
import streamlit.components.v1 as components
|
5 |
from langchain import OpenAI
|
6 |
from langchain.callbacks import get_openai_callback
|
7 |
from langchain.chains import ConversationChain
|
8 |
from langchain.chains.conversation.memory import ConversationSummaryMemory
|
9 |
+
from message import Message
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
|
11 |
|
12 |
def load_css():
|
message.py
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from dataclasses import dataclass
|
2 |
+
from typing import Literal
|
3 |
+
|
4 |
+
|
5 |
+
@dataclass
|
6 |
+
class Message:
|
7 |
+
"""Class for keeping track of a chat message."""
|
8 |
+
|
9 |
+
origin: Literal["human", "ai"]
|
10 |
+
message: str
|