chore: gamic

main.py

@@ -9,7 +9,7 @@ os.environ['TRANSFORMERS_CACHE'] = '/tmp'
 from telegram.ext import (
     Application,
 )
-from api import GirlfriendGPT
+from api import BuddyPanda
 from functools import partial  
 from typing import List
 from telegram import Update
@@ -41,12 +41,12 @@ class LoggingDisabled:
         logging.disable(logging.NOTSET)
 
 
-def main():
+def main() -> None:
     app = Application.builder().token(
         '5998527257:AAFolKz-AIXDzaSj3UOmxaLZSyDVp_3LoYw',).build()
 
     run_agent(
-        agent=GirlfriendGPT(
+        agent=BuddyPanda(
             token="hello",
             application=app
         )
@@ -54,7 +54,7 @@ def main():
     print(f"Starting Agent...")
 
 
-def run_agent(agent: GirlfriendGPT, as_api: bool = False) -> None:
+def run_agent(agent: BuddyPanda, as_api: bool = False) -> None:
     agent.handlers()
 
 

src/agent/base.py

@@ -3,7 +3,6 @@ import re
 from abc import abstractmethod
 
 from telegram import Update
-import telegram
 from telegram.ext import (
     ConversationHandler,
     CallbackContext,
@@ -15,7 +14,7 @@ from agent.tools.conversation import Conversation
 SELECT_COMMAND, GET_TEXT = range(2)
 
 
-class LangChainAgentBot(ConversationSummary,  Conversation):
+class BuddyAgentBot(ConversationSummary,  Conversation):
     def is_verbose_logging_enabled(self):
         return True
 

src/agent/tools/conversation.py

@@ -1,7 +1,6 @@
 import logging
 from telegram import Update
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
 from telegram.ext import (
     CallbackContext,
 )

src/api.py

@@ -4,12 +4,11 @@ from typing import List, Optional, Type
 from pydantic import Field
 
 
-from agent.base import LangChainAgentBot
-from telegram.ext import Updater, CommandHandler, CallbackContext, Application, ContextTypes
+from agent.base import BuddyAgentBot
+from telegram.ext import CommandHandler, CallbackContext, Application, ContextTypes
 from telegram import Update
 from telegram.ext import (
     CommandHandler,
-    ConversationHandler,
     CallbackContext,
     MessageHandler,
     filters
@@ -24,7 +23,7 @@ async def hello(update: Update, context: CallbackContext) -> None:
     await update.message.reply_text(intro_text)
 
 
-class GirlfriendGPT(LangChainAgentBot, ):
+class BuddyPanda(BuddyAgentBot, ):
     """Deploy LangChain chatbots and connect them to Telegram."""
 
     token: str
@@ -45,4 +44,4 @@ class GirlfriendGPT(LangChainAgentBot, ):
         self.application.add_handler(summary_handler)
         self.application.add_handler(CommandHandler('start', hello))
 
-        # self.application.run_polling()
+        self.application.run_polling()