chore: started

main.py

@@ -43,7 +43,7 @@ class LoggingDisabled:
 
 def main():
     app = Application.builder().token(
-        '6207542226:AAFhEmSjjhJbzjocoxE3FcgFHhz_rTaLXbU',).build()
+        '6207542226:AAF1Lir2JjjFfux8KYyAzOEzzTjUIhPB4bI',).build()
 
     run_agent(
         agent=GirlfriendGPT(

src/agent/tools/conversation.py

@@ -20,12 +20,14 @@ class Conversation():
     tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
     model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
+    chat_history_ids = torch.tensor([], dtype=torch.long)
+
     async def talk(self, message: str):
         logging.info(f"{message}")
         new_user_input_ids = self.tokenizer(message + self.tokenizer.eos_token, return_tensors='pt')
-        bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1)
-        chat_history_ids =self.model.generate(bot_input_ids, max_length=1000, pad_token_id=self.tokenizer.eos_token_id)
-        return "{}".format(self.tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True))
+        bot_input_ids = torch.cat([self.chat_history_ids, new_user_input_ids], dim=-1)
+        self.chat_history_ids =self.model.generate(bot_input_ids, max_length=1000, pad_token_id=self.tokenizer.eos_token_id)
+        return "{}".format(self.tokenizer.decode(self.chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True))
 
     
     async def process_conversation(self, update: Update, context: CallbackContext) -> int: