remove ipython functions
Browse files- src/few_shot_funcs.py +0 -14
src/few_shot_funcs.py
CHANGED
@@ -5,7 +5,6 @@ import inflect
|
|
5 |
import pandas as pd
|
6 |
from typing import Dict
|
7 |
from datasets import load_dataset
|
8 |
-
from IPython.display import display, HTML
|
9 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
10 |
from langchain.vectorstores import FAISS
|
11 |
from langchain.vectorstores.utils import DistanceStrategy
|
@@ -52,19 +51,6 @@ class Conversation:
|
|
52 |
|
53 |
message = {"role": role, "content": content}
|
54 |
self.messages.append(message)
|
55 |
-
|
56 |
-
def display_conversation(self) -> None:
|
57 |
-
SEP = '\n'
|
58 |
-
for message in self.messages:
|
59 |
-
if message['role'] == 'system':
|
60 |
-
display(HTML(f'<b>{message["content"]}</b>'))
|
61 |
-
elif message['role'] == 'user':
|
62 |
-
msg_align = message["content"].replace("Title:", "<br><b>Title:</b>")\
|
63 |
-
.replace("Technical details:", "<br><b>Technical details:</b>").replace("### Feature-bullets:", "<br><b>Feature-bullets:</b>")
|
64 |
-
display(HTML(f'<p style="background-color:White; color:Black; padding:5px;">{msg_align}</p>'))
|
65 |
-
else:
|
66 |
-
msg_align = message["content"].lstrip(SEP).replace(SEP, "<br><br>")
|
67 |
-
display(HTML(f'<p style="background-color:LightGray; color:Black; padding:5px;">{msg_align}</p>'))
|
68 |
|
69 |
def api_call(messages: Dict[str, str], temperature: float = 0.7, top_p: int = 1, n_responses: int = 1) -> dict:
|
70 |
"""
|
|
|
5 |
import pandas as pd
|
6 |
from typing import Dict
|
7 |
from datasets import load_dataset
|
|
|
8 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
9 |
from langchain.vectorstores import FAISS
|
10 |
from langchain.vectorstores.utils import DistanceStrategy
|
|
|
51 |
|
52 |
message = {"role": role, "content": content}
|
53 |
self.messages.append(message)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
|
55 |
def api_call(messages: Dict[str, str], temperature: float = 0.7, top_p: int = 1, n_responses: int = 1) -> dict:
|
56 |
"""
|