leo-bourrel commited on
Commit
be06203
·
1 Parent(s): 6a2d3ac

feat: split message class

Browse files
Files changed (2) hide show
  1. app.py +3 -12
  2. 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
- import streamlit.components.v1 as components
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