Spaces:
Runtime error
Runtime error
Commit
·
ac8786f
1
Parent(s):
67c7151
chore: added information
Browse files- main.py +11 -12
- src/agent/tools/text_summary.py +0 -2
- src/api.py +9 -0
main.py
CHANGED
@@ -37,9 +37,7 @@ logger = logging.getLogger(__name__)
|
|
37 |
|
38 |
|
39 |
|
40 |
-
|
41 |
-
intro_text = f"🤖 Greetings human!🤗\nI'm a bot built by Rexthecoder\n🦾 I can do a lot of things"
|
42 |
-
await update.message.reply_text(intro_text)
|
43 |
|
44 |
|
45 |
# async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
|
@@ -76,16 +74,17 @@ def main():
|
|
76 |
|
77 |
|
78 |
def run_agent(agent: GirlfriendGPT, as_api: bool = False) -> None:
|
|
|
79 |
# For Debugging
|
80 |
-
summary_handler = agent.conversation_summary_handler()
|
81 |
-
agent.application.add_handler(summary_handler)
|
82 |
-
agent.application.add_handler(CommandHandler('start', hello))
|
83 |
-
# agent.application.add_handler(CommandHandler('summary', agent.conversation_summary))
|
84 |
-
# agent.application.add_handler(MessageHandler(
|
85 |
-
# filters.TEXT & ~filters.COMMAND, agent.create_response))
|
86 |
-
agent.application.run_polling()
|
87 |
-
# agent.application.add_handler(MessageHandler(
|
88 |
-
# filters.TEXT & ~filters.COMMAND, agent.create_response))
|
89 |
|
90 |
|
91 |
if __name__ == "__main__":
|
|
|
37 |
|
38 |
|
39 |
|
40 |
+
|
|
|
|
|
41 |
|
42 |
|
43 |
# async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
|
|
|
74 |
|
75 |
|
76 |
def run_agent(agent: GirlfriendGPT, as_api: bool = False) -> None:
|
77 |
+
agent.handlers()
|
78 |
# For Debugging
|
79 |
+
# summary_handler = agent.conversation_summary_handler()
|
80 |
+
# agent.application.add_handler(summary_handler)
|
81 |
+
# agent.application.add_handler(CommandHandler('start', hello))
|
82 |
+
# # agent.application.add_handler(CommandHandler('summary', agent.conversation_summary))
|
83 |
+
# # agent.application.add_handler(MessageHandler(
|
84 |
+
# # filters.TEXT & ~filters.COMMAND, agent.create_response))
|
85 |
+
# agent.application.run_polling()
|
86 |
+
# # agent.application.add_handler(MessageHandler(
|
87 |
+
# # filters.TEXT & ~filters.COMMAND, agent.create_response))
|
88 |
|
89 |
|
90 |
if __name__ == "__main__":
|
src/agent/tools/text_summary.py
CHANGED
@@ -35,9 +35,7 @@ class ConversationSummary():
|
|
35 |
"mrm8488/flan-t5-small-finetuned-samsum")
|
36 |
|
37 |
async def summarize(self, input: str, words: int):
|
38 |
-
|
39 |
logging.info(f"{input} {words}")
|
40 |
-
|
41 |
input_ids = self.tokenizer(input, return_tensors="pt").input_ids
|
42 |
outputs = self.model.generate(input_ids, max_length=words)
|
43 |
decoded_output = self.tokenizer.decode(
|
|
|
35 |
"mrm8488/flan-t5-small-finetuned-samsum")
|
36 |
|
37 |
async def summarize(self, input: str, words: int):
|
|
|
38 |
logging.info(f"{input} {words}")
|
|
|
39 |
input_ids = self.tokenizer(input, return_tensors="pt").input_ids
|
40 |
outputs = self.model.generate(input_ids, max_length=words)
|
41 |
decoded_output = self.tokenizer.decode(
|
src/api.py
CHANGED
@@ -20,6 +20,9 @@ class GirlFriendAIConfig():
|
|
20 |
default="", description="Optional voice_id for ElevenLabs Voice Bot"
|
21 |
)
|
22 |
|
|
|
|
|
|
|
23 |
|
24 |
class GirlfriendGPT(LangChainAgentBot):
|
25 |
"""Deploy LangChain chatbots and connect them to Telegram."""
|
@@ -34,6 +37,12 @@ class GirlfriendGPT(LangChainAgentBot):
|
|
34 |
# Run the bot until the user presses Ctrl-C
|
35 |
# self.application.run_polling()
|
36 |
self.token = token
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
|
38 |
# async def echo(self, update: Update, context: CallbackContext) -> None:
|
39 |
# """Echo the user message."""
|
|
|
20 |
default="", description="Optional voice_id for ElevenLabs Voice Bot"
|
21 |
)
|
22 |
|
23 |
+
async def hello(update: Update, context: CallbackContext) -> None:
|
24 |
+
intro_text = f"🤖 Greetings human!🤗\nI'm a bot built by Rexthecoder\n🦾 I can do a lot of things"
|
25 |
+
await update.message.reply_text(intro_text)
|
26 |
|
27 |
class GirlfriendGPT(LangChainAgentBot):
|
28 |
"""Deploy LangChain chatbots and connect them to Telegram."""
|
|
|
37 |
# Run the bot until the user presses Ctrl-C
|
38 |
# self.application.run_polling()
|
39 |
self.token = token
|
40 |
+
|
41 |
+
def handlers(self):
|
42 |
+
summary_handler = self.conversation_summary_handler()
|
43 |
+
self.application.add_handler(summary_handler)
|
44 |
+
self.application.add_handler(CommandHandler('start', hello))
|
45 |
+
self.application.run_polling()
|
46 |
|
47 |
# async def echo(self, update: Update, context: CallbackContext) -> None:
|
48 |
# """Echo the user message."""
|