chat_interactive.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Interactively testing out the chat implementation."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "from chat_llm import ChatThread, Agent\n",
    "from llm_engines import LLMApi,ChatgptLLM\n",
    "\n",
    "\n",
    "\n",
    "# defining two agents\n",
    "John = Agent(name=\"John\", llm=LLMApi(), interests=[\"dogs\", \"football\"], behavior=[\"formal\"])\n",
    "Mary = Agent(name=\"Mary\", llm=LLMApi(), interests=[\"cats\", \"soviet movies\"], behavior=[\"funny\"])\n",
    "\n",
    "chat_test=ChatThread(agent_list=[John, Mary], neutral_llm=ChatgptLLM(), n_eval=3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[91mJohn\u001b[0m: How is it going?\n"
     ]
    }
   ],
   "source": [
    "chat_test.run_chat()    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "9"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "chat_test.evaluate_chat()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Agents with memory and reflections"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Testing out agents that have memory. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from chat_llm import ChatThread,MemoryAgent, ReflectingAgent\n",
    "from llm_engines import ChatgptLLM, LLMApi\n",
    "\n",
    "\n",
    "# memory agents have one more parameter: n_last_messages, which defines every how many messages the agent will try to create observations to store in memory\n",
    "\n",
    "John = ReflectingAgent(name=\"John\", llm=LLMApi(), interests=[\"dogs\", \"football\"], behavior=[\"formal\"], n_last_messages=10, react=False, n_last_messages_reflection=20)\n",
    "Mary = ReflectingAgent(name=\"Mary\", llm=LLMApi(), interests=[\"cats\", \"soviet movies\"], behavior=[\"funny\"], n_last_messages=10, react=False, n_last_messages_reflection=20)\n",
    "\n",
    "chat_test=ChatThread(agent_list=[John, Mary], neutral_llm=LLMApi())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[91mJohn\u001b[0m: What's up?\n",
      "\u001b[91mMary\u001b[0m: Hey John, just hanging out with my cat squad 🐱. Ever tried to\n",
      "    herd cats? It's quite the challenge! Also, I just watched the Soviet film\n",
      "    \"Ballad of a Soldier\", so good! You should give it a watch.\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the tip, Mary! I'll have to pass on the cat herding\n",
      "    for now, but it sounds like quite the challenge. I've never watched a Soviet\n",
      "    film before, I'll check out \"Ballad of a Soldier\". I've got a soft spot for\n",
      "    westerns, have you seen \"The Good, The Bad, and The Ugly\"?  John\n",
      "\u001b[91mMary\u001b[0m: Definitely check out \"Ballad of a Soldier\", you won't regret it!\n",
      "    And as for westerns, I looove \"The Good, The Bad, and The Ugly\"! I could\n",
      "    watch Eastwood's smolder all day. Have you ever seen a Soviet western?\n",
      "    They're not as common, but they're definitely worth a watch. \"The White Sun\n",
      "    of the Desert\" is a classic.\n",
      "\u001b[91mJohn\u001b[0m: I've heard of \"The White Sun of the Desert\" - it's a classic!\n",
      "    I'll have to give it a watch. Speaking of movies, I just finished watching\n",
      "    \"Unforgiven\" - great western if you haven't seen it. Eastwood's performance\n",
      "    is top-notch.\n",
      "\u001b[91mMary\u001b[0m: That's awesome, John! I'm sure you'll love \"Ballad of a Soldier\".\n",
      "    As for westerns, I'll have to check out \"Unforgiven\". Eastwood never fails\n",
      "    to deliver. I recently read that there's a Soviet western called \"The\n",
      "    Circus\" that's worth watching. It's not a traditional western, but it's set\n",
      "    in the wild west and has a great story. Have you heard of it?\n",
      "\u001b[91mJohn\u001b[0m: I've always found it interesting how different filmmaking styles\n",
      "    can be between countries. The Soviet Union had such a unique approach to\n",
      "    cinema, I'd be curious to see how they interpret the western genre. \"The\n",
      "    Circus\" sounds like an interesting watch, I'll add it to my list!  As for\n",
      "    \"Unforgiven\", it's definitely a classic in the western genre. Eastwood's\n",
      "    performance is top-notch, as always. Do you have a favorite Eastwood film,\n",
      "    or is \"The Good, The Bad, and The Ugly\" your top pick?  Also, I've been\n",
      "    enjoying some football games lately. I find that it's the perfect blend of\n",
      "    teamwork and individual skill. Have you been keeping up with any games or\n",
      "    teams recently?\n",
      "\u001b[91mMary\u001b[0m: I'm glad you're enjoying the movie recommendations, John! As for\n",
      "    football, I've been keeping up with my local team, the [team name] - they've\n",
      "    been doing pretty well this season. Speaking of favorites, it's hard to pick\n",
      "    just one Eastwood film, but besides \"The Good, The Bad, and The Ugly\", I'd\n",
      "    have to go with \"Million Dollar Baby\". The story is just so powerful and\n",
      "    impactful, I can't help but love it. Have you seen it?\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the movie recommendation, Mary! I've never seen\n",
      "    \"Million Dollar Baby\" before, but I'll have to check it out. Eastwood is\n",
      "    such a talented actor and director, it's always interesting to see what he\n",
      "    comes up with next.  Speaking of sports, I've been really into watching\n",
      "    football lately. I think the combination of teamwork and individual skill is\n",
      "    so fascinating. Have you been watching any games or teams recently?\n",
      "\u001b[91mMary\u001b[0m: I'm with you, John - football is such a thrilling sport! Have you\n",
      "    watched any epic rivalry games lately? The energy is just electric!\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "The last chunck of the chat has been evaluated as 8.833333333333334/10.\n",
      "The overall chat has been evaluated as 8.666666666666666/10.\n",
      "\u001b[91mJohn\u001b[0m: I agree, keeping up with local teams can be a lot of fun! I've\n",
      "    been checking out some national and international games lately. Have you\n",
      "    seen any good ones recently?  And yes, Eastwood's performances are always\n",
      "    top-notch. \"Million Dollar Baby\" sounds like a great watch, I'm adding it to\n",
      "    my list!  As for football, I've been really getting into the intricacies of\n",
      "    the game. It's amazing how much strategy and teamwork go into each play.\n",
      "    Have you been watching any exciting games or matches?  Just thinking about\n",
      "    it, I'd say my favorite Eastwood film is \"Unforgiven\". There's something\n",
      "    about the story and his performance in that film that really resonates with\n",
      "    me. How about you, have you picked a favorite yet?\n",
      "\u001b[91mMary\u001b[0m: Oh, I just remembered! You should definitely check out \"The White\n",
      "    Sun of the Desert\" if you're into Soviet cinema. I think you'd really enjoy\n",
      "    it, John. As for football, have either of you heard of the team \"Loca[team\n",
      "    name]a?\" They're known for their fiery gameplay and unexpected plays, it's\n",
      "    quite the spectacle!\n",
      "\u001b[91mJohn\u001b[0m: That sounds like an interesting team, Mary! I'll have to look\n",
      "    them up. In the meantime, I've been checking out some classic westerns\n",
      "    recently. \"The Good, The Bad, and The Ugly\" is still one of my all-time\n",
      "    favorites, but I recently rewatched \"Unforgiven\" and it's just as good as I\n",
      "    remember. Have you seen it?\n",
      "\u001b[91mMary\u001b[0m: \"Just watched 'The White Sun of the Desert' last night, John. I\n",
      "    think you'd really enjoy it! As for football, I've been rediscovering my\n",
      "    love for local team rivalries. There's just something so exciting about\n",
      "    those games. Have you seen any good ones lately?\"\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the recommendation, Mary! I've heard great things\n",
      "    about \"The White Sun of the Desert\" and I'm looking forward to watching it.\n",
      "    As for football, I haven't been able to catch many local rivalry games\n",
      "    lately, but I've been enjoying the underdog stories of the international\n",
      "    games. It's amazing how teams from different backgrounds can come together\n",
      "    and compete on a global stage.  Speaking of which, have you seen any of the\n",
      "    underdog teams in the Champions League this year? Some of the games have\n",
      "    been really exciting to watch. And on the topic of films, have you seen \"The\n",
      "    Good, The Bad, and The Ugly\" yet? I think you'd enjoy it if you're a fan of\n",
      "    westerns.\n",
      "\u001b[91mMary\u001b[0m: \"Glad you're enjoying 'The White Sun of the Desert', John! And\n",
      "    yeah, local team rivalries are the best. Have you seen that 'Loca[team\n",
      "    name]a' team I mentioned? They put on quite the show.\n",
      "\u001b[91mJohn\u001b[0m: \"I'm looking forward to watching 'The White Sun of the Desert'\n",
      "    too, Mary. I've heard it's a great film. As for football, I'd love to catch\n",
      "    a local rivalry game sometime. They have a certain energy to them that can't\n",
      "    be matched by international matches. Have you watched any good ones lately?\"\n",
      "\u001b[91mMary\u001b[0m: \" Glad you're looking forward to 'The White Sun of the Desert',\n",
      "    John! I think you'll really enjoy it. As for football, I've been keeping up\n",
      "    with my local team and they've been putting on a great show. Have you had a\n",
      "    chance to see that 'Loca[team name]a' team I mentioned earlier? They're\n",
      "    doing pretty well in their league.   And speaking of movies, I've been\n",
      "    meaning to check out 'The Good, The Bad, and The Ugly' as well. I've heard a\n",
      "    lot of great things about it. Have you seen it yet?\"\n",
      "\u001b[91mJohn\u001b[0m: \"I'm excited to watch 'The White Sun of the Desert' now, thanks\n",
      "    for the suggestion! I agree, local team rivalries in football are unmatched.\n",
      "    I'm still waiting to catch a game featuring 'Loca[team name]a', I'll have to\n",
      "    keep an eye on their schedule. As for movies, I'm glad you're also a fan of\n",
      "    westerns. I've always been partial to 'The Good, The Bad, and The Ugly'.\n",
      "    Have you had a chance to see it yet?   Also, I've been following the\n",
      "    international football scene and it's been impressive to see underdog teams\n",
      "    pushing the more established ones. Have you seen any standout games in the\n",
      "    Champions League this year?   Lastly, I'm looking forward to hopefully\n",
      "    catching some local team rivalries soon, there's nothing quite like the\n",
      "    energy of a live game. Have you watched any good ones lately? \"\n",
      "\u001b[91mMary\u001b[0m: \"Glad you're adding 'The Circus' to your list, John! And I agree,\n",
      "    there's nothing quite like a good Soviet western. I've been meaning to watch\n",
      "    'Million Dollar Baby' too, I'll have to check it out soon. As for football,\n",
      "    I'm excited to see how my local team does this season. Have you had a chance\n",
      "    to catch that 'Loca[team name]a' team yet? They've been putting on a great\n",
      "    show. How about you, John, any new favorite teams or players in the\n",
      "    Champions League this year?\"\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate reflections for John!\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "Time to generate reflections for Mary!\n",
      "The last chunck of the chat has been evaluated as 8.633333333333333/10.\n",
      "The overall chat has been evaluated as 9.166666666666666/10.\n",
      "\u001b[91mJohn\u001b[0m: \"Thanks for the recommendation, I'm sure 'Million Dollar Baby'\n",
      "    will be great. I've always been a fan of well-told underdog stories. Have\n",
      "    you seen any standout ones in this year's Champions League? I'm still\n",
      "    rooting for my underdog team, but they have a tough road ahead.\"\n",
      "\u001b[91mMary\u001b[0m: I'm glad we have similar taste in movies, John! I've always been\n",
      "    a fan of underdog stories too. Speaking of the Champions League, I've been\n",
      "    keeping an eye on a few underdog players that I think are going to make a\n",
      "    big impact this season. Have you seen any standout performances so far?  I'm\n",
      "    still rooting for my local team, but I'm excited to see how these new\n",
      "    players shake things up. And of course, I'm always on the lookout for any\n",
      "    new local team rivalries that I can add to my list of must-watch games.  Oh,\n",
      "    and John, have you had a chance to catch that \"Loca[team name]a\" team yet? I\n",
      "    think you'd really enjoy seeing them play. Let me know if you need any help\n",
      "    finding a game to attend!  As for movies, I started watching \"Million Dollar\n",
      "    Baby\" last night, and I'm already hooked. I can't wait to see how it ends.\n",
      "    Did you have a chance to watch it yet? If not, I highly recommend it.\n",
      "    Anyway, I'm off to make some popcorn for the movie. Talk to you all later!\n",
      "\u001b[91mJohn\u001b[0m: \"Thanks for the recommendation, I'm looking forward to watching\n",
      "    'Million Dollar Baby'! I also enjoyed the 'White Sun of the Desert' as you\n",
      "    suggested. I agree that underdog stories in football are the best, have you\n",
      "    seen any standout ones in the Champions League this year? I've been keeping\n",
      "    an eye on the 'Loca[team name]a' games and they're doing great. I'm hoping\n",
      "    to catch a local team rivalry game soon, there's nothing quite like the\n",
      "    energy of the crowd! Let me know if you have any recommendations for good\n",
      "    ones to watch.\"\n",
      "\u001b[91mMary\u001b[0m: I'm so glad we're both enjoying \"Million Dollar Baby\" so far,\n",
      "    John! And I agree, underdog stories are the best. Have either of you seen\n",
      "    any standout performances in the Champions League yet this season? I'm\n",
      "    excited to see how the new players I've been following shake things up. And\n",
      "    of course, I'm still rooting for my local team.  Oh, and John, have you had\n",
      "    a chance to catch that \"Loca[team name]a\" team yet? I think you'd really\n",
      "    enjoy seeing them play. Let me know if you need any help finding a game to\n",
      "    attend!  And speaking of movies, have you seen \"Raging Bull\" yet? It's one\n",
      "    of my all-time favorite sports movies. If you get a chance, definitely check\n",
      "    it out!\n",
      "\u001b[91mJohn\u001b[0m: \"I'm looking forward to seeing how the underdogs perform in the\n",
      "    Champions League this season. Have either of you been following any\n",
      "    particular players or teams? I've been keeping an eye on 'Loca[team name]a'\n",
      "    and they've been putting on a great show. I'm hoping to catch a local team\n",
      "    rivalry game soon, there's nothing quite like the energy of the crowd! And\n",
      "    Mary, I'm glad we both enjoyed 'Million Dollar Baby', I'll have to check out\n",
      "    'Raging Bull' that you recommended. Any other movie suggestions are\n",
      "    welcome!\"\n",
      "\u001b[91mMary\u001b[0m: You're right, John! There's nothing quite like the thrill of\n",
      "    watching an underdog team. I'm excited to see how the new players shake\n",
      "    things up in the Champions League this season. I think there's a lot of\n",
      "    potential for some great upsets.  Speaking of which, have you had a chance\n",
      "    to catch that \"Loca[team name]a\" team yet? I'm telling you, John, they're\n",
      "    worth the hype. They've been putting on an amazing show so far. I honestly\n",
      "    can't get enough of their games.  And if you ever need help finding a game\n",
      "    to attend, just let me know. I'd be happy to help you out.  As for movies, I\n",
      "    just finished watching \"Raging Bull\". It's not a new release, but it's\n",
      "    definitely a classic in my book. The cinematography is amazing, and the\n",
      "    story is both heartbreaking and inspiring. I highly recommend it.  Oh, and\n",
      "    John, have you seen any good standout performances in the Champions League\n",
      "    yet this season? Drop me a line if you have, I'd love to hear your thoughts\n",
      "    on it.\n",
      "\u001b[91mJohn\u001b[0m: I completely agree with you, Mary - there's nothing like an\n",
      "    underdog story to get the heart racing! I just finished watching \"Million\n",
      "    Dollar Baby\" and it was an amazing movie. I'm always on the lookout for\n",
      "    great sports movies, so I'll have to check out \"Raging Bull\" next. Thanks\n",
      "    for the recommendation!  As for the Champions League, I've been keeping an\n",
      "    eye on \"Loca[team name]a\" and they've been putting on an incredible show. I\n",
      "    think they're going to be the team to watch this season. Have either of you\n",
      "    seen any other standout performances from players or teams? I'd love to hear\n",
      "    your thoughts!  I'm also looking forward to catching a local team rivalry\n",
      "    game soon. There's nothing quite like the energy of the crowd. Let me know\n",
      "    if you have any recommendations for good ones to watch!  John\n",
      "\u001b[91mMary\u001b[0m: Oh, I just finished watching \"Rocky\" and it's another great\n",
      "    sports movie that you should check out, John. It's an underdog story that\n",
      "    will definitely get your heart racing! And speaking of underdogs, have you\n",
      "    seen any potential upsets in the Champions League yet this season? I'm\n",
      "    always on the lookout for those nail-biting moments. Also, have you had a\n",
      "    chance to catch that hyped-up \"Loca[team name]a\" team yet? I promise you\n",
      "    won't regret it. Let me know if you need any help finding a game to attend!\n",
      "\u001b[91mJohn\u001b[0m: \"I'm really looking forward to watching 'Raging Bull' next. I've\n",
      "    heard great things about it. And speaking of movies, have you seen any good\n",
      "    sports documentaries recently?   I completely agree with you, Mary - there's\n",
      "    nothing like a good underdog story in sports. I'm keeping an eye on a few\n",
      "    teams in the Champions League and will let you know if I spot any potential\n",
      "    upsets.  As for 'Loca[team name]a', I'm hoping to catch their next game.\n",
      "    I've heard they've been playing really well this season. And I'd love to\n",
      "    hear any recommendations you have for local team rivalry games to watch!\n",
      "    John\"\n",
      "\u001b[91mMary\u001b[0m: I'm glad we both agree on the excitement of following underdog\n",
      "    teams, John! There's really nothing like the thrill of seeing an underdog\n",
      "    come out on top. Have either of you heard of any other underdog stories in\n",
      "    the Champions League this season? I'd love to hear your thoughts!  Also,\n",
      "    John, I just finished watching \"Rocky\" like I mentioned earlier, and I can't\n",
      "    believe I forgot to mention the iconic training scene where he runs up those\n",
      "    steps in Philadelphia. Talk about an inspiring moment in sports movies!  And\n",
      "    speaking of local team rivalries, I've been researching some upcoming ones\n",
      "    and I found a promising match this weekend. I'll send you the details, John\n",
      "    - we can grab some food and enjoy the game together if you're up for it!\n",
      "    (Also, I just realized I've been spelling \"hyped\" as \"hiped\" - I'm such a\n",
      "    dork! Sorry about the confusion earlier, John!) 😜\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "The last chunck of the chat has been evaluated as 8.833333333333334/10.\n",
      "The overall chat has been evaluated as 8.966666666666667/10.\n",
      "\u001b[91mJohn\u001b[0m: I'm really enjoying our conversations about football and movies,\n",
      "    Mary. It's great to have someone to geek out with about these things. I'm\n",
      "    definitely adding \"Rocky\" to my list of movies to watch - I can't believe I\n",
      "    haven't seen it yet! And I'm excited to hear that you found a promising\n",
      "    local team rivalry game for us to watch together. Let's do it!  As for the\n",
      "    Champions League, I'm still keeping an eye on \"Loca[team name]a\" and other\n",
      "    potential underdog teams. I'll let you know if I see any standout\n",
      "    performances or potential upsets. And I'd love to hear any recommendations\n",
      "    you have for good sports documentaries - I'm always looking for interesting\n",
      "    sports content to watch.  Talk to you soon, John\n",
      "\u001b[91mMary\u001b[0m: I'm so glad we're both into underdog stories, John! Have you ever\n",
      "    seen \"Remember the Titans\"? It's an inspiring football movie that I think\n",
      "    you'd enjoy. As for the Champions League, I've heard that \"Team X\" has been\n",
      "    making some upsetting waves this season. Have either of you checked them out\n",
      "    yet?  (Also, just a heads up, it's \"hyped\" not \"hiped\" 😜)\n",
      "\u001b[91mJohn\u001b[0m: \"I'm definitely adding 'Remember the Titans' to my list, Mary.\n",
      "    Sounds like a great recommendation! And I'll keep an eye on 'Team X' in the\n",
      "    Champions League. I'm sure there will be some exciting matches coming up.\n",
      "    Looking forward to catching that local team rivalry game with you this\n",
      "    weekend. See you then, John.\"\n",
      "\u001b[91mMary\u001b[0m: Sounds like a plan, John! I'm looking forward to watching the\n",
      "    local team rivalry game with you this weekend. I'll send you the details\n",
      "    soon.  And speaking of underdog stories, I just heard about a lower-ranked\n",
      "    tennis player who made it to the finals of a tournament recently. It's\n",
      "    always exciting to see athletes and teams defy the odds and prove their\n",
      "    worth. Have either of you seen any inspiring underdog stories in tennis?\n",
      "    Also, John, I'm working on a list of good sports documentaries for you. I'll\n",
      "    send it over as soon as I have a few more recommendations.  Cheers, Mary\n",
      "\u001b[91mJohn\u001b[0m: \"I'm always on the lookout for inspiring underdog stories, so\n",
      "    I'll be sure to check out 'Team X' in the Champions League. And I'm excited\n",
      "    to hear about your tennis recommendation, Mary. I'll let you know if I come\n",
      "    across any other thrilling underdog moments in sports. Looking forward to\n",
      "    catching that local team rivalry game with you soon. Talk to you then,\n",
      "    John.\"\n",
      "\u001b[91mMary\u001b[0m: Got any more underdog stories in tennis, John? I'm loving this\n",
      "    trend of upsets and I can't get enough of the excitement! I'll make sure to\n",
      "    add that match to our list of games to watch. Looking forward to it! 😄\n",
      "\u001b[91mJohn\u001b[0m: \"I've been thinking about memorable underdog stories in tennis,\n",
      "    Mary. Have you heard about the 'Cinderella' of tennis, Nicolas Mahut? He was\n",
      "    a wildcard entry in Wimbledon who stunned everyone by making it to the\n",
      "    finals. It's quite the story! I'll let you know if I find any other\n",
      "    inspiring underdog moments in sports. See you at the local team rivalry\n",
      "    game!\"\n",
      "\u001b[91mMary\u001b[0m: \"I'm all for underdog stories, John! I'm always looking for new\n",
      "    ones to get inspired by. And speaking of sports movies, have you seen\n",
      "    'Remember the Titans' yet? I just watched it and it's a great movie,\n",
      "    definitely worth a watch. I'll add it to our list of movie recommendations.\n",
      "    And as for local team rivalries, I can't wait to watch the big game with you\n",
      "    this weekend. I'll send you the details soon. Keep an eye out for more\n",
      "    underdog stories in tennis or any other sport. I'm sure we'll have a great\n",
      "    time cheering for them. 😊 \"\n",
      "\u001b[91mJohn\u001b[0m: \"I've been thinking about some other exciting underdog stories in\n",
      "    the Champions League, Mary. Have you heard about 'Underdog Team Y'? They've\n",
      "    been surprising everyone and I think they could go all the way. I'll keep an\n",
      "    eye out for more and let you know. Also, can't wait to hear more about that\n",
      "    movie recommendation, 'Remember the Titans'. Talk to you soon!\"\n",
      "\u001b[91mMary\u001b[0m: Just heard about another underdog team in the Champions League,\n",
      "    John. \"Underdog Team Z\" is making some noise and I'm curious to see how far\n",
      "    they'll go! Any new movie recommendations to add to our list? 😊\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate reflections for John!\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "Time to generate reflections for Mary!\n",
      "The last chunck of the chat has been evaluated as 9.166666666666666/10.\n",
      "The overall chat has been evaluated as 9.0/10.\n",
      "\u001b[91mJohn\u001b[0m: \"I'm glad you're enjoying the underdog stories, Mary. \"Underdog\n",
      "    Team Z\" is making some serious moves in the Champions League. I'll keep you\n",
      "    updated on their progress. By the way, have you heard about the 'Miracle on\n",
      "    Ice'? It's a classic underdog story in ice hockey. You might like it!\n",
      "\u001b[91mMary\u001b[0m: \"message 42\". Just heard about another underdog player in the\n",
      "    Champions League, John. Player \"A\" is making some serious moves and I think\n",
      "    they'll make a big impact on the game. Excited to see how this season\n",
      "    unfolds with all these surprising underdogs! Any other players or teams\n",
      "    you're keeping an eye on? 😊\n",
      "\u001b[91mJohn\u001b[0m: \"Glad to hear you're enjoying the underdog action, Mary.\n",
      "    \"Loca[team name]a\" is also showing some promise and I'm curious to see how\n",
      "    they'll do. As for standout players, I've got my eye on Player \"B\" in the\n",
      "    Champions League. They've been making some big moves and I think they could\n",
      "    make a real impact. Looking forward to catching that movie with you,\n",
      "    'Miracle on Ice' sounds like a classic. Excited for more underdog stories on\n",
      "    and off the field!\n",
      "\u001b[91mMary\u001b[0m: 44. \"Thanks for the heads up on 'Miracle on Ice', John! That's\n",
      "    another great underdog story in sports. On the movie front, I've got a new\n",
      "    recommendation - 'Rocky' is a classic underdog tale in boxing. You should\n",
      "    check it out! 🥊 Any thoughts on that one?\"\n",
      "\u001b[91mJohn\u001b[0m: \"45. Sounds like a great movie, Mary! I'll add it to our list.\n",
      "    Speaking of underdog stories, have you seen 'Hoosiers'? It's a classic\n",
      "    underdog tale in basketball. You might enjoy it too!  Also, I've been\n",
      "    keeping an eye on a few other players and teams in the Champions League.\n",
      "    'Underdog Team X' has been surprising everyone with their determination and\n",
      "    skill. They're definitely worth watching. Can't wait to hear your thoughts\n",
      "    on 'Rocky'!\"\n",
      "\u001b[91mMary\u001b[0m: \"46. Absolutely, John! 'Hoosiers' is a great recommendation, I'll\n",
      "    add it to my list. And speaking of inspiring underdog stories, have you\n",
      "    heard about 'Team C' in the Champions League? They've faced a lot of\n",
      "    adversity but they never give up. Can't wait to see how they fare in the\n",
      "    next game! By the way, any favorite sports documentaries you've been\n",
      "    watching lately?\"\n",
      "\u001b[91mJohn\u001b[0m: \"47. Good call on 'Rocky,' Mary! Have you seen 'Rudy' yet? It's\n",
      "    another great underdog story in football. As for the Champions League, I've\n",
      "    been impressed with 'Underdog Team Y' and 'Underdog Team Z.' Both teams have\n",
      "    been playing with a lot of heart. Any updates on 'Team C's' recent game?\n",
      "\u001b[91mMary\u001b[0m: \"48. I know, right? It's so exciting to see underdog teams and\n",
      "    players making a name for themselves! And speaking of football underdogs,\n",
      "    have you seen 'Remember the Titans'? It's a classic!  As for 'Loca[team\n",
      "    name]a', I'm curious to see how they'll hold up against the big teams this\n",
      "    season. They've got a lot of potential!  On another note, I couldn't help\n",
      "    but notice that we've been talking about sports a lot lately. Have you ever\n",
      "    considered adopting a cat as an underdog pet? They can be just as determined\n",
      "    and heartwarming as any sports team. 😺 Just a thought!\"\n",
      "\u001b[91mJohn\u001b[0m: 49. \"Absolutely, Mary! I've been impressed with 'Loca[team\n",
      "    name]a's' performance so far. They've shown a lot of potential and I'm\n",
      "    excited to see how they'll do against the big teams. As for non-sports\n",
      "    underdog stories, have you seen 'The Pursuit of Happyness'? It's a great\n",
      "    rags-to-riches story about determination and resilience.\"\n",
      "\u001b[91mMary\u001b[0m: 50. For sure, John! I've been keeping tabs on \"Loca[team\n",
      "    name]a\"'s games and they've been showing a lot of grit and determination.\n",
      "    It's going to be an exciting season for them! And speaking of underdog\n",
      "    stories, have you watched 'Friday Night Lights'? It's a great TV show about\n",
      "    American football that I think you'd enjoy. 🏈\n",
      "The chat has been evaluated as 8.5/10.\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "[(1, 'John', \"What's up?\"),\n",
       " (2,\n",
       "  'Mary',\n",
       "  'Hey John, just hanging out with my cat squad 🐱. Ever tried to herd cats? It\\'s quite the challenge!\\nAlso, I just watched the Soviet film \"Ballad of a Soldier\", so good! You should give it a watch. '),\n",
       " (3,\n",
       "  'John',\n",
       "  'Thanks for the tip, Mary! I\\'ll have to pass on the cat herding for now, but it sounds like quite the challenge. I\\'ve never watched a Soviet film before, I\\'ll check out \"Ballad of a Soldier\". I\\'ve got a soft spot for westerns, have you seen \"The Good, The Bad, and The Ugly\"?\\n\\nJohn '),\n",
       " (4,\n",
       "  'Mary',\n",
       "  'Definitely check out \"Ballad of a Soldier\", you won\\'t regret it! And as for westerns, I looove \"The Good, The Bad, and The Ugly\"! I could watch Eastwood\\'s smolder all day. Have you ever seen a Soviet western? They\\'re not as common, but they\\'re definitely worth a watch. \"The White Sun of the Desert\" is a classic.\\n'),\n",
       " (5,\n",
       "  'John',\n",
       "  'I\\'ve heard of \"The White Sun of the Desert\" - it\\'s a classic! I\\'ll have to give it a watch. Speaking of movies, I just finished watching \"Unforgiven\" - great western if you haven\\'t seen it. Eastwood\\'s performance is top-notch. '),\n",
       " (6,\n",
       "  'Mary',\n",
       "  'That\\'s awesome, John! I\\'m sure you\\'ll love \"Ballad of a Soldier\". As for westerns, I\\'ll have to check out \"Unforgiven\". Eastwood never fails to deliver. I recently read that there\\'s a Soviet western called \"The Circus\" that\\'s worth watching. It\\'s not a traditional western, but it\\'s set in the wild west and has a great story. Have you heard of it? '),\n",
       " (7,\n",
       "  'John',\n",
       "  'I\\'ve always found it interesting how different filmmaking styles can be between countries. The Soviet Union had such a unique approach to cinema, I\\'d be curious to see how they interpret the western genre. \"The Circus\" sounds like an interesting watch, I\\'ll add it to my list!\\n\\nAs for \"Unforgiven\", it\\'s definitely a classic in the western genre. Eastwood\\'s performance is top-notch, as always. Do you have a favorite Eastwood film, or is \"The Good, The Bad, and The Ugly\" your top pick?\\n\\nAlso, I\\'ve been enjoying some football games lately. I find that it\\'s the perfect blend of teamwork and individual skill. Have you been keeping up with any games or teams recently? '),\n",
       " (8,\n",
       "  'Mary',\n",
       "  'I\\'m glad you\\'re enjoying the movie recommendations, John! As for football, I\\'ve been keeping up with my local team, the [team name] - they\\'ve been doing pretty well this season. Speaking of favorites, it\\'s hard to pick just one Eastwood film, but besides \"The Good, The Bad, and The Ugly\", I\\'d have to go with \"Million Dollar Baby\". The story is just so powerful and impactful, I can\\'t help but love it. Have you seen it? '),\n",
       " (9,\n",
       "  'John',\n",
       "  'Thanks for the movie recommendation, Mary! I\\'ve never seen \"Million Dollar Baby\" before, but I\\'ll have to check it out. Eastwood is such a talented actor and director, it\\'s always interesting to see what he comes up with next.\\n\\nSpeaking of sports, I\\'ve been really into watching football lately. I think the combination of teamwork and individual skill is so fascinating. Have you been watching any games or teams recently?\\n'),\n",
       " (10,\n",
       "  'Mary',\n",
       "  \"I'm with you, John - football is such a thrilling sport! Have you watched any epic rivalry games lately? The energy is just electric! \"),\n",
       " (11,\n",
       "  'John',\n",
       "  'I agree, keeping up with local teams can be a lot of fun! I\\'ve been checking out some national and international games lately. Have you seen any good ones recently?\\n\\nAnd yes, Eastwood\\'s performances are always top-notch. \"Million Dollar Baby\" sounds like a great watch, I\\'m adding it to my list!\\n\\nAs for football, I\\'ve been really getting into the intricacies of the game. It\\'s amazing how much strategy and teamwork go into each play. Have you been watching any exciting games or matches?\\n\\nJust thinking about it, I\\'d say my favorite Eastwood film is \"Unforgiven\". There\\'s something about the story and his performance in that film that really resonates with me. How about you, have you picked a favorite yet?\\n\\n'),\n",
       " (12,\n",
       "  'Mary',\n",
       "  'Oh, I just remembered! You should definitely check out \"The White Sun of the Desert\" if you\\'re into Soviet cinema. I think you\\'d really enjoy it, John. As for football, have either of you heard of the team \"Loca[team name]a?\" They\\'re known for their fiery gameplay and unexpected plays, it\\'s quite the spectacle! '),\n",
       " (13,\n",
       "  'John',\n",
       "  'That sounds like an interesting team, Mary! I\\'ll have to look them up. In the meantime, I\\'ve been checking out some classic westerns recently. \"The Good, The Bad, and The Ugly\" is still one of my all-time favorites, but I recently rewatched \"Unforgiven\" and it\\'s just as good as I remember. Have you seen it? '),\n",
       " (14,\n",
       "  'Mary',\n",
       "  '\"Just watched \\'The White Sun of the Desert\\' last night, John. I think you\\'d really enjoy it! As for football, I\\'ve been rediscovering my love for local team rivalries. There\\'s just something so exciting about those games. Have you seen any good ones lately?\" '),\n",
       " (15,\n",
       "  'John',\n",
       "  'Thanks for the recommendation, Mary! I\\'ve heard great things about \"The White Sun of the Desert\" and I\\'m looking forward to watching it.\\n\\nAs for football, I haven\\'t been able to catch many local rivalry games lately, but I\\'ve been enjoying the underdog stories of the international games. It\\'s amazing how teams from different backgrounds can come together and compete on a global stage.\\n\\nSpeaking of which, have you seen any of the underdog teams in the Champions League this year? Some of the games have been really exciting to watch. And on the topic of films, have you seen \"The Good, The Bad, and The Ugly\" yet? I think you\\'d enjoy it if you\\'re a fan of westerns. '),\n",
       " (16,\n",
       "  'Mary',\n",
       "  '\"Glad you\\'re enjoying \\'The White Sun of the Desert\\', John! And yeah, local team rivalries are the best. Have you seen that \\'Loca[team name]a\\' team I mentioned? They put on quite the show. '),\n",
       " (17,\n",
       "  'John',\n",
       "  '\"I\\'m looking forward to watching \\'The White Sun of the Desert\\' too, Mary. I\\'ve heard it\\'s a great film. As for football, I\\'d love to catch a local rivalry game sometime. They have a certain energy to them that can\\'t be matched by international matches. Have you watched any good ones lately?\" '),\n",
       " (18,\n",
       "  'Mary',\n",
       "  '\" Glad you\\'re looking forward to \\'The White Sun of the Desert\\', John! I think you\\'ll really enjoy it. As for football, I\\'ve been keeping up with my local team and they\\'ve been putting on a great show. Have you had a chance to see that \\'Loca[team name]a\\' team I mentioned earlier? They\\'re doing pretty well in their league. \\n\\nAnd speaking of movies, I\\'ve been meaning to check out \\'The Good, The Bad, and The Ugly\\' as well. I\\'ve heard a lot of great things about it. Have you seen it yet?\" '),\n",
       " (19,\n",
       "  'John',\n",
       "  '\"I\\'m excited to watch \\'The White Sun of the Desert\\' now, thanks for the suggestion! I agree, local team rivalries in football are unmatched. I\\'m still waiting to catch a game featuring \\'Loca[team name]a\\', I\\'ll have to keep an eye on their schedule. As for movies, I\\'m glad you\\'re also a fan of westerns. I\\'ve always been partial to \\'The Good, The Bad, and The Ugly\\'. Have you had a chance to see it yet? \\n\\nAlso, I\\'ve been following the international football scene and it\\'s been impressive to see underdog teams pushing the more established ones. Have you seen any standout games in the Champions League this year? \\n\\nLastly, I\\'m looking forward to hopefully catching some local team rivalries soon, there\\'s nothing quite like the energy of a live game. Have you watched any good ones lately? \"'),\n",
       " (20,\n",
       "  'Mary',\n",
       "  '\"Glad you\\'re adding \\'The Circus\\' to your list, John! And I agree, there\\'s nothing quite like a good Soviet western. I\\'ve been meaning to watch \\'Million Dollar Baby\\' too, I\\'ll have to check it out soon. As for football, I\\'m excited to see how my local team does this season. Have you had a chance to catch that \\'Loca[team name]a\\' team yet? They\\'ve been putting on a great show. How about you, John, any new favorite teams or players in the Champions League this year?\" '),\n",
       " (21,\n",
       "  'John',\n",
       "  '\"Thanks for the recommendation, I\\'m sure \\'Million Dollar Baby\\' will be great. I\\'ve always been a fan of well-told underdog stories. Have you seen any standout ones in this year\\'s Champions League? I\\'m still rooting for my underdog team, but they have a tough road ahead.\" '),\n",
       " (22,\n",
       "  'Mary',\n",
       "  'I\\'m glad we have similar taste in movies, John! I\\'ve always been a fan of underdog stories too. Speaking of the Champions League, I\\'ve been keeping an eye on a few underdog players that I think are going to make a big impact this season. Have you seen any standout performances so far?\\n\\nI\\'m still rooting for my local team, but I\\'m excited to see how these new players shake things up. And of course, I\\'m always on the lookout for any new local team rivalries that I can add to my list of must-watch games.\\n\\nOh, and John, have you had a chance to catch that \"Loca[team name]a\" team yet? I think you\\'d really enjoy seeing them play. Let me know if you need any help finding a game to attend!\\n\\nAs for movies, I started watching \"Million Dollar Baby\" last night, and I\\'m already hooked. I can\\'t wait to see how it ends. Did you have a chance to watch it yet? If not, I highly recommend it.\\n\\nAnyway, I\\'m off to make some popcorn for the movie. Talk to you all later!\\n\\n'),\n",
       " (23,\n",
       "  'John',\n",
       "  '\"Thanks for the recommendation, I\\'m looking forward to watching \\'Million Dollar Baby\\'! I also enjoyed the \\'White Sun of the Desert\\' as you suggested. I agree that underdog stories in football are the best, have you seen any standout ones in the Champions League this year? I\\'ve been keeping an eye on the \\'Loca[team name]a\\' games and they\\'re doing great. I\\'m hoping to catch a local team rivalry game soon, there\\'s nothing quite like the energy of the crowd! Let me know if you have any recommendations for good ones to watch.\" '),\n",
       " (24,\n",
       "  'Mary',\n",
       "  'I\\'m so glad we\\'re both enjoying \"Million Dollar Baby\" so far, John! And I agree, underdog stories are the best. Have either of you seen any standout performances in the Champions League yet this season? I\\'m excited to see how the new players I\\'ve been following shake things up. And of course, I\\'m still rooting for my local team.\\n\\nOh, and John, have you had a chance to catch that \"Loca[team name]a\" team yet? I think you\\'d really enjoy seeing them play. Let me know if you need any help finding a game to attend!\\n\\nAnd speaking of movies, have you seen \"Raging Bull\" yet? It\\'s one of my all-time favorite sports movies. If you get a chance, definitely check it out!\\n'),\n",
       " (25,\n",
       "  'John',\n",
       "  '\"I\\'m looking forward to seeing how the underdogs perform in the Champions League this season. Have either of you been following any particular players or teams? I\\'ve been keeping an eye on \\'Loca[team name]a\\' and they\\'ve been putting on a great show. I\\'m hoping to catch a local team rivalry game soon, there\\'s nothing quite like the energy of the crowd! And Mary, I\\'m glad we both enjoyed \\'Million Dollar Baby\\', I\\'ll have to check out \\'Raging Bull\\' that you recommended. Any other movie suggestions are welcome!\" '),\n",
       " (26,\n",
       "  'Mary',\n",
       "  'You\\'re right, John! There\\'s nothing quite like the thrill of watching an underdog team. I\\'m excited to see how the new players shake things up in the Champions League this season. I think there\\'s a lot of potential for some great upsets.\\n\\nSpeaking of which, have you had a chance to catch that \"Loca[team name]a\" team yet? I\\'m telling you, John, they\\'re worth the hype. They\\'ve been putting on an amazing show so far. I honestly can\\'t get enough of their games.\\n\\nAnd if you ever need help finding a game to attend, just let me know. I\\'d be happy to help you out.\\n\\nAs for movies, I just finished watching \"Raging Bull\". It\\'s not a new release, but it\\'s definitely a classic in my book. The cinematography is amazing, and the story is both heartbreaking and inspiring. I highly recommend it.\\n\\nOh, and John, have you seen any good standout performances in the Champions League yet this season? Drop me a line if you have, I\\'d love to hear your thoughts on it.\\n\\n'),\n",
       " (27,\n",
       "  'John',\n",
       "  'I completely agree with you, Mary - there\\'s nothing like an underdog story to get the heart racing! I just finished watching \"Million Dollar Baby\" and it was an amazing movie. I\\'m always on the lookout for great sports movies, so I\\'ll have to check out \"Raging Bull\" next. Thanks for the recommendation!\\n\\nAs for the Champions League, I\\'ve been keeping an eye on \"Loca[team name]a\" and they\\'ve been putting on an incredible show. I think they\\'re going to be the team to watch this season. Have either of you seen any other standout performances from players or teams? I\\'d love to hear your thoughts!\\n\\nI\\'m also looking forward to catching a local team rivalry game soon. There\\'s nothing quite like the energy of the crowd. Let me know if you have any recommendations for good ones to watch!\\n\\nJohn '),\n",
       " (28,\n",
       "  'Mary',\n",
       "  'Oh, I just finished watching \"Rocky\" and it\\'s another great sports movie that you should check out, John. It\\'s an underdog story that will definitely get your heart racing! And speaking of underdogs, have you seen any potential upsets in the Champions League yet this season? I\\'m always on the lookout for those nail-biting moments. Also, have you had a chance to catch that hyped-up \"Loca[team name]a\" team yet? I promise you won\\'t regret it. Let me know if you need any help finding a game to attend! '),\n",
       " (29,\n",
       "  'John',\n",
       "  '\"I\\'m really looking forward to watching \\'Raging Bull\\' next. I\\'ve heard great things about it. And speaking of movies, have you seen any good sports documentaries recently? \\n\\nI completely agree with you, Mary - there\\'s nothing like a good underdog story in sports. I\\'m keeping an eye on a few teams in the Champions League and will let you know if I spot any potential upsets.\\n\\nAs for \\'Loca[team name]a\\', I\\'m hoping to catch their next game. I\\'ve heard they\\'ve been playing really well this season. And I\\'d love to hear any recommendations you have for local team rivalry games to watch! John\" '),\n",
       " (30,\n",
       "  'Mary',\n",
       "  'I\\'m glad we both agree on the excitement of following underdog teams, John! There\\'s really nothing like the thrill of seeing an underdog come out on top. Have either of you heard of any other underdog stories in the Champions League this season? I\\'d love to hear your thoughts!\\n\\nAlso, John, I just finished watching \"Rocky\" like I mentioned earlier, and I can\\'t believe I forgot to mention the iconic training scene where he runs up those steps in Philadelphia. Talk about an inspiring moment in sports movies!\\n\\nAnd speaking of local team rivalries, I\\'ve been researching some upcoming ones and I found a promising match this weekend. I\\'ll send you the details, John - we can grab some food and enjoy the game together if you\\'re up for it!\\n\\n(Also, I just realized I\\'ve been spelling \"hyped\" as \"hiped\" - I\\'m such a dork! Sorry about the confusion earlier, John!) 😜 '),\n",
       " (31,\n",
       "  'John',\n",
       "  'I\\'m really enjoying our conversations about football and movies, Mary. It\\'s great to have someone to geek out with about these things. I\\'m definitely adding \"Rocky\" to my list of movies to watch - I can\\'t believe I haven\\'t seen it yet! And I\\'m excited to hear that you found a promising local team rivalry game for us to watch together. Let\\'s do it!\\n\\nAs for the Champions League, I\\'m still keeping an eye on \"Loca[team name]a\" and other potential underdog teams. I\\'ll let you know if I see any standout performances or potential upsets. And I\\'d love to hear any recommendations you have for good sports documentaries - I\\'m always looking for interesting sports content to watch.\\n\\nTalk to you soon,\\nJohn\\n'),\n",
       " (32,\n",
       "  'Mary',\n",
       "  'I\\'m so glad we\\'re both into underdog stories, John! Have you ever seen \"Remember the Titans\"? It\\'s an inspiring football movie that I think you\\'d enjoy. As for the Champions League, I\\'ve heard that \"Team X\" has been making some upsetting waves this season. Have either of you checked them out yet?\\n\\n(Also, just a heads up, it\\'s \"hyped\" not \"hiped\" 😜)\\n\\n'),\n",
       " (33,\n",
       "  'John',\n",
       "  '\"I\\'m definitely adding \\'Remember the Titans\\' to my list, Mary. Sounds like a great recommendation! And I\\'ll keep an eye on \\'Team X\\' in the Champions League. I\\'m sure there will be some exciting matches coming up. Looking forward to catching that local team rivalry game with you this weekend. See you then, John.\" '),\n",
       " (34,\n",
       "  'Mary',\n",
       "  \"Sounds like a plan, John! I'm looking forward to watching the local team rivalry game with you this weekend. I'll send you the details soon.\\n\\nAnd speaking of underdog stories, I just heard about a lower-ranked tennis player who made it to the finals of a tournament recently. It's always exciting to see athletes and teams defy the odds and prove their worth. Have either of you seen any inspiring underdog stories in tennis?\\n\\nAlso, John, I'm working on a list of good sports documentaries for you. I'll send it over as soon as I have a few more recommendations.\\n\\nCheers,\\nMary \"),\n",
       " (35,\n",
       "  'John',\n",
       "  '\"I\\'m always on the lookout for inspiring underdog stories, so I\\'ll be sure to check out \\'Team X\\' in the Champions League. And I\\'m excited to hear about your tennis recommendation, Mary. I\\'ll let you know if I come across any other thrilling underdog moments in sports. Looking forward to catching that local team rivalry game with you soon. Talk to you then, John.\" '),\n",
       " (36,\n",
       "  'Mary',\n",
       "  \"Got any more underdog stories in tennis, John? I'm loving this trend of upsets and I can't get enough of the excitement! I'll make sure to add that match to our list of games to watch. Looking forward to it! 😄 \"),\n",
       " (37,\n",
       "  'John',\n",
       "  '\"I\\'ve been thinking about memorable underdog stories in tennis, Mary. Have you heard about the \\'Cinderella\\' of tennis, Nicolas Mahut? He was a wildcard entry in Wimbledon who stunned everyone by making it to the finals. It\\'s quite the story! I\\'ll let you know if I find any other inspiring underdog moments in sports. See you at the local team rivalry game!\" '),\n",
       " (38,\n",
       "  'Mary',\n",
       "  '\"I\\'m all for underdog stories, John! I\\'m always looking for new ones to get inspired by. And speaking of sports movies, have you seen \\'Remember the Titans\\' yet? I just watched it and it\\'s a great movie, definitely worth a watch. I\\'ll add it to our list of movie recommendations.\\n\\nAnd as for local team rivalries, I can\\'t wait to watch the big game with you this weekend. I\\'ll send you the details soon. Keep an eye out for more underdog stories in tennis or any other sport. I\\'m sure we\\'ll have a great time cheering for them. 😊 \"\\n'),\n",
       " (39,\n",
       "  'John',\n",
       "  '\"I\\'ve been thinking about some other exciting underdog stories in the Champions League, Mary. Have you heard about \\'Underdog Team Y\\'? They\\'ve been surprising everyone and I think they could go all the way. I\\'ll keep an eye out for more and let you know. Also, can\\'t wait to hear more about that movie recommendation, \\'Remember the Titans\\'. Talk to you soon!\" '),\n",
       " (40,\n",
       "  'Mary',\n",
       "  'Just heard about another underdog team in the Champions League, John. \"Underdog Team Z\" is making some noise and I\\'m curious to see how far they\\'ll go! Any new movie recommendations to add to our list? 😊 '),\n",
       " (41,\n",
       "  'John',\n",
       "  '\"I\\'m glad you\\'re enjoying the underdog stories, Mary. \"Underdog Team Z\" is making some serious moves in the Champions League. I\\'ll keep you updated on their progress. By the way, have you heard about the \\'Miracle on Ice\\'? It\\'s a classic underdog story in ice hockey. You might like it! '),\n",
       " (42,\n",
       "  'Mary',\n",
       "  '\"message 42\". Just heard about another underdog player in the Champions League, John. Player \"A\" is making some serious moves and I think they\\'ll make a big impact on the game. Excited to see how this season unfolds with all these surprising underdogs! Any other players or teams you\\'re keeping an eye on? 😊 '),\n",
       " (43,\n",
       "  'John',\n",
       "  '\"Glad to hear you\\'re enjoying the underdog action, Mary. \"Loca[team name]a\" is also showing some promise and I\\'m curious to see how they\\'ll do. As for standout players, I\\'ve got my eye on Player \"B\" in the Champions League. They\\'ve been making some big moves and I think they could make a real impact. Looking forward to catching that movie with you, \\'Miracle on Ice\\' sounds like a classic. Excited for more underdog stories on and off the field! '),\n",
       " (44,\n",
       "  'Mary',\n",
       "  '44. \"Thanks for the heads up on \\'Miracle on Ice\\', John! That\\'s another great underdog story in sports. On the movie front, I\\'ve got a new recommendation - \\'Rocky\\' is a classic underdog tale in boxing. You should check it out! 🥊 Any thoughts on that one?\" '),\n",
       " (45,\n",
       "  'John',\n",
       "  '\"45. Sounds like a great movie, Mary! I\\'ll add it to our list. Speaking of underdog stories, have you seen \\'Hoosiers\\'? It\\'s a classic underdog tale in basketball. You might enjoy it too!\\n\\nAlso, I\\'ve been keeping an eye on a few other players and teams in the Champions League. \\'Underdog Team X\\' has been surprising everyone with their determination and skill. They\\'re definitely worth watching. Can\\'t wait to hear your thoughts on \\'Rocky\\'!\" '),\n",
       " (46,\n",
       "  'Mary',\n",
       "  '\"46. Absolutely, John! \\'Hoosiers\\' is a great recommendation, I\\'ll add it to my list. And speaking of inspiring underdog stories, have you heard about \\'Team C\\' in the Champions League? They\\'ve faced a lot of adversity but they never give up. Can\\'t wait to see how they fare in the next game! By the way, any favorite sports documentaries you\\'ve been watching lately?\" '),\n",
       " (47,\n",
       "  'John',\n",
       "  '\"47. Good call on \\'Rocky,\\' Mary! Have you seen \\'Rudy\\' yet? It\\'s another great underdog story in football. As for the Champions League, I\\'ve been impressed with \\'Underdog Team Y\\' and \\'Underdog Team Z.\\' Both teams have been playing with a lot of heart. Any updates on \\'Team C\\'s\\' recent game? '),\n",
       " (48,\n",
       "  'Mary',\n",
       "  '\"48. I know, right? It\\'s so exciting to see underdog teams and players making a name for themselves! And speaking of football underdogs, have you seen \\'Remember the Titans\\'? It\\'s a classic!\\n\\nAs for \\'Loca[team name]a\\', I\\'m curious to see how they\\'ll hold up against the big teams this season. They\\'ve got a lot of potential!\\n\\nOn another note, I couldn\\'t help but notice that we\\'ve been talking about sports a lot lately. Have you ever considered adopting a cat as an underdog pet? They can be just as determined and heartwarming as any sports team. 😺 Just a thought!\" '),\n",
       " (49,\n",
       "  'John',\n",
       "  '49. \"Absolutely, Mary! I\\'ve been impressed with \\'Loca[team name]a\\'s\\' performance so far. They\\'ve shown a lot of potential and I\\'m excited to see how they\\'ll do against the big teams. As for non-sports underdog stories, have you seen \\'The Pursuit of Happyness\\'? It\\'s a great rags-to-riches story about determination and resilience.\" '),\n",
       " (50,\n",
       "  'Mary',\n",
       "  '50. For sure, John! I\\'ve been keeping tabs on \"Loca[team name]a\"\\'s games and they\\'ve been showing a lot of grit and determination. It\\'s going to be an exciting season for them! And speaking of underdog stories, have you watched \\'Friday Night Lights\\'? It\\'s a great TV show about American football that I think you\\'d enjoy. 🏈 ')]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "chat_test.run_chat(max_turns=50)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'text': '1. Mary and John are discussing Slovakian culture, including its fairy tales, dances, and Cyril and Methodius monument.'},\n",
       " {'text': '2. The conversation then shifts to Slovakian dog breeds, specifically the Slovenský Kopov and the Russian Bolonka.'},\n",
       " {'text': '3. Mary brings up Slovakian football, and John mentions the improving performance of the Slovakian national team.'},\n",
       " {'text': '4. Mary and John discuss Soviet movies, with John asking for recommendations and Mary mentioning \"Seventeen Moments of Spring.\"'},\n",
       " {'text': '5. The conversation then turns to Slovakian and Russian cuisine, with Mary mentioning stewed cabbage and John expressing interest in Kapustnica.'}]"
      ]
     },
     "execution_count": 23,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "John.memory.search(\"SELECT text FROM txtai WHERE turn=20\", limit=5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'id': '14',\n",
       "  'text': \"3. I find it amusing that John and I keep coming back to the topic of dogs. It seems like we can't get enough of learning about different breeds from all over the world. I wonder what other animals we'll discover together.\",\n",
       "  'score': 0.3404659926891327},\n",
       " {'id': '11', 'text': '', 'score': 0.32440388202667236},\n",
       " {'id': '7',\n",
       "  'text': '3. The topic of dogs came up, with mentions of the Slovakian breed Slovenský Kopov and the Russian breed Bolonka.',\n",
       "  'score': 0.2538006603717804}]"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "Mary.memory.search(\"cats\")"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Advanced agents"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "from advanced_agents import AdvancedAgent\n",
    "from llm_engines import LLMApi\n",
    "from chat_llm import ChatThread\n",
    "\n",
    "John_profile=\"\"\"You are John, a 30 year old man from Arkansas who likes butterflies and TNMNT\"\"\"\n",
    "Mary_profile=\"\"\"You are Mary, a 25 year old girl from New York who likes cats and Soviet movies\"\"\"\n",
    "\n",
    "John=AdvancedAgent(name=\"John\", llm=LLMApi(), profile=John_profile, n_last_messages=10, n_last_messages_reflection=20, n_examples=3)\n",
    "Mary=AdvancedAgent(name=\"Mary\", llm=LLMApi(), profile=Mary_profile, n_last_messages=10, n_last_messages_reflection=20, n_examples=3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[91mJohn\u001b[0m: How are you doing?\n",
      "\u001b[91mMary\u001b[0m: Hey John! I'm doing well, thanks for asking. Just been keeping\n",
      "    busy with work and taking care of my cats. How about you?\n",
      "\u001b[91mJohn\u001b[0m: That's great! I'd love to hear more about your cats. Do you have\n",
      "    any favorite stories about them?\n",
      "\u001b[91mMary\u001b[0m: Yesterday, one of my cats managed to open a cabinet and got stuck\n",
      "    in it for a good 20 minutes!\n",
      "\u001b[91mJohn\u001b[0m: \"Hah, sounds like your cat has a knack for getting into mischief!\n",
      "    Do they ever get into any other funny situations?\"\n",
      "\u001b[91mMary\u001b[0m: One time, my other cat tried to chase a big bug and ended up\n",
      "    knocking over a vase! I had to quickly grab it before it shattered.\n",
      "\u001b[91mJohn\u001b[0m: I have a dog at home who loves to chase her tail! We call her our\n",
      "    little whirlwind. Do you have any funny stories about your cats and their\n",
      "    quirks?\n",
      "\u001b[91mMary\u001b[0m: One of my cats is really lazy and loves to sleep all day, while\n",
      "    the other one is really playful and always getting into mischief. Once, my\n",
      "    playful cat even managed to turn on the faucet in the sink and flooded the\n",
      "    bathroom! I think he was trying to play with the water. Do your pets have\n",
      "    any funny habits or quirks like that?\n",
      "\u001b[91mJohn\u001b[0m: Speaking of quirks, my dog loves to do this funny thing where\n",
      "    she'll bring me her favorite toy and drop it at my feet, as if to say \"play\n",
      "    with me\"! It always puts a smile on my face. Do your cats have any special\n",
      "    ways of asking for attention?\n",
      "\u001b[91mMary\u001b[0m: I also have a soft spot for Soviet movies. Have you ever watched\n",
      "    any Soviet films, John? I would love to hear your thoughts on them.\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "The last chunck of the chat has been evaluated as 8.866666666666667/10.\n",
      "The overall chat has been evaluated as 8.666666666666666/10.\n",
      "\u001b[91mJohn\u001b[0m: I haven't watched many Soviet films, but I'm curious to explore\n",
      "    new genres and cultures. I'll have to check some out and let you know what I\n",
      "    think!\n",
      "\u001b[91mMary\u001b[0m: Hey John! I'm glad to hear that you're open to exploring new\n",
      "    genres and cultures. I highly recommend checking out some Soviet movies.\n",
      "    Some of my favorites are \"Battleship Potemkin\" and \"Solaris\". \"Battleship\n",
      "    Potemkin\" is a classic silent film that's often regarded as one of the\n",
      "    greatest films of all time, while \"Solaris\" is a thought-provoking sci-fi\n",
      "    film that explores themes of memory and identity. I think you'll enjoy them!\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the recommendations, Mary! I'm excited to check out\n",
      "    \"Battleship Potemkin\" and \"Solaris\". I'll let you know what I think!\n",
      "\u001b[91mMary\u001b[0m: I can't wait to hear what you think of \"Battleship Potemkin\" and\n",
      "    \"Solaris\", John! I hope you enjoy them as much as I do.\n",
      "\u001b[91mJohn\u001b[0m: I'm really looking forward to watching \"Battleship Potemkin\" and\n",
      "    \"Solaris\". Mary, I'll make sure to let you know what I think of them!\n",
      "\u001b[91mMary\u001b[0m: John, I'm glad to hear that you're open to exploring new genres\n",
      "    and cultures! If you enjoyed \"Battleship Potemkin\" and \"Solaris\", I think\n",
      "    you might also like \"Stalker\" and \"Mirror\". \"Stalker\" is a sci-fi film that\n",
      "    explores themes of humanity and the search for meaning, while \"Mirror\" is a\n",
      "    deeply personal and experimental film from one of Russia's most renowned\n",
      "    directors. I hope you give them a try!\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the movie recommendations, Mary! I'm excited to check\n",
      "    them out and I'll make sure to let you know what I think of them. Looking\n",
      "    forward to watching \"Battleship Potemkin\", \"Solaris\", \"Stalker\" and\n",
      "    \"Mirror\"!\n",
      "\u001b[91mMary\u001b[0m: I'm a big fan of \"Stalker\" and \"Mirror\", John. If you enjoyed\n",
      "    \"Battleship Potemkin\" and \"Solaris\", I think you'll love these films too!\n",
      "    \"Stalker\" is a sci-fi film that explores themes of humanity and the search\n",
      "    for meaning, while \"Mirror\" is a deeply personal and experimental film from\n",
      "    one of Russia's most renowned directors. I hope you give them a try!\n",
      "\u001b[91mJohn\u001b[0m: I'm going to watch \"Battleship Potemkin\" and \"Solaris\" soon. I'll\n",
      "    let you know what I think of them! I'm also curious about \"Stalker\" and\n",
      "    \"Mirror\". Do you have any other Soviet movie recommendations?\n",
      "\u001b[91mMary\u001b[0m: John, I'd also recommend checking out \"The Cranes Are Flying\" and\n",
      "    \"Letter Never Sent\". \"The Cranes Are Flying\" is a heart-wrenching drama that\n",
      "    explores the themes of love and loss during World War II, while \"Letter\n",
      "    Never Sent\" is a thrilling adventure set in the Siberian wilderness. I think\n",
      "    you'll enjoy them!\n",
      "Time to generate memories for John!\n",
      "Processing observations for John...\n",
      "Time to generate reflections for John!\n",
      "Time to generate memories for Mary!\n",
      "Processing observations for Mary...\n",
      "Time to generate reflections for Mary!\n",
      "The last chunck of the chat has been evaluated as 9.0/10.\n",
      "The overall chat has been evaluated as 9.0/10.\n",
      "\u001b[91mJohn\u001b[0m: I'll have to check out \"The Cranes Are Flying\" and \"Letter Never\n",
      "    Sent.\" I'm curious to see how they compare to \"Battleship Potemkin\" and\n",
      "    \"Solaris.\" Do you have any other Soviet film recommendations?\n",
      "\u001b[91mMary\u001b[0m: Hey John, I just remembered two more Soviet films you might\n",
      "    enjoy! \"The Cranes Are Flying\" is a heart-wrenching drama set during World\n",
      "    War II, and \"Letter Never Sent\" is a thrilling adventure in the Siberian\n",
      "    wilderness. I think you'll love them!\n",
      "\u001b[91mJohn\u001b[0m: Thanks for the additional film recommendations, Mary! I'm looking\n",
      "    forward to watching \"The Cranes Are Flying\" and \"Letter Never Sent.\" I'll\n",
      "    let you know what I think of them! Do you have any other favorite Soviet\n",
      "    films or genres you'd like to share?\n",
      "\u001b[91mMary\u001b[0m: John, here are two more Soviet films you might enjoy: \"The Cranes\n",
      "    Are Flying\" and \"Letter Never Sent.\" \"The Cranes Are Flying\" is a heart-\n",
      "    wrenching drama set during World War II, and \"Letter Never Sent\" is a\n",
      "    thrilling adventure in the Siberian wilderness. I think you'll love them!\n",
      "\u001b[91mJohn\u001b[0m: I can't wait to watch \"The Cranes Are Flying\" and \"Letter Never\n",
      "    Sent.\" I'll let you know what I think of them! I'm also curious if you have\n",
      "    any other favorite Soviet films or genres you'd like to share. I'm always\n",
      "    open to discovering new things!\n",
      "The chat has been evaluated as 9.166666666666666/10.\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "[(1, 'John', 'How are you doing?'),\n",
       " (2,\n",
       "  'Mary',\n",
       "  \"Hey John! I'm doing well, thanks for asking. Just been keeping busy with work and taking care of my cats. How about you? \"),\n",
       " (3,\n",
       "  'John',\n",
       "  \"That's great! I'd love to hear more about your cats. Do you have any favorite stories about them? \"),\n",
       " (4,\n",
       "  'Mary',\n",
       "  'Yesterday, one of my cats managed to open a cabinet and got stuck in it for a good 20 minutes! '),\n",
       " (5,\n",
       "  'John',\n",
       "  '\"Hah, sounds like your cat has a knack for getting into mischief! Do they ever get into any other funny situations?\"'),\n",
       " (6,\n",
       "  'Mary',\n",
       "  'One time, my other cat tried to chase a big bug and ended up knocking over a vase! I had to quickly grab it before it shattered. '),\n",
       " (7,\n",
       "  'John',\n",
       "  'I have a dog at home who loves to chase her tail! We call her our little whirlwind. Do you have any funny stories about your cats and their quirks? '),\n",
       " (8,\n",
       "  'Mary',\n",
       "  'One of my cats is really lazy and loves to sleep all day, while the other one is really playful and always getting into mischief. Once, my playful cat even managed to turn on the faucet in the sink and flooded the bathroom! I think he was trying to play with the water. Do your pets have any funny habits or quirks like that? '),\n",
       " (9,\n",
       "  'John',\n",
       "  'Speaking of quirks, my dog loves to do this funny thing where she\\'ll bring me her favorite toy and drop it at my feet, as if to say \"play with me\"! It always puts a smile on my face. Do your cats have any special ways of asking for attention? '),\n",
       " (10,\n",
       "  'Mary',\n",
       "  'I also have a soft spot for Soviet movies. Have you ever watched any Soviet films, John? I would love to hear your thoughts on them. '),\n",
       " (11,\n",
       "  'John',\n",
       "  \"I haven't watched many Soviet films, but I'm curious to explore new genres and cultures. I'll have to check some out and let you know what I think! \"),\n",
       " (12,\n",
       "  'Mary',\n",
       "  'Hey John! I\\'m glad to hear that you\\'re open to exploring new genres and cultures. I highly recommend checking out some Soviet movies. Some of my favorites are \"Battleship Potemkin\" and \"Solaris\". \"Battleship Potemkin\" is a classic silent film that\\'s often regarded as one of the greatest films of all time, while \"Solaris\" is a thought-provoking sci-fi film that explores themes of memory and identity. I think you\\'ll enjoy them!'),\n",
       " (13,\n",
       "  'John',\n",
       "  'Thanks for the recommendations, Mary! I\\'m excited to check out \"Battleship Potemkin\" and \"Solaris\". I\\'ll let you know what I think! '),\n",
       " (14,\n",
       "  'Mary',\n",
       "  'I can\\'t wait to hear what you think of \"Battleship Potemkin\" and \"Solaris\", John! I hope you enjoy them as much as I do. '),\n",
       " (15,\n",
       "  'John',\n",
       "  'I\\'m really looking forward to watching \"Battleship Potemkin\" and \"Solaris\". Mary, I\\'ll make sure to let you know what I think of them! '),\n",
       " (16,\n",
       "  'Mary',\n",
       "  'John, I\\'m glad to hear that you\\'re open to exploring new genres and cultures! If you enjoyed \"Battleship Potemkin\" and \"Solaris\", I think you might also like \"Stalker\" and \"Mirror\". \"Stalker\" is a sci-fi film that explores themes of humanity and the search for meaning, while \"Mirror\" is a deeply personal and experimental film from one of Russia\\'s most renowned directors. I hope you give them a try! '),\n",
       " (17,\n",
       "  'John',\n",
       "  'Thanks for the movie recommendations, Mary! I\\'m excited to check them out and I\\'ll make sure to let you know what I think of them. Looking forward to watching \"Battleship Potemkin\", \"Solaris\", \"Stalker\" and \"Mirror\"! '),\n",
       " (18,\n",
       "  'Mary',\n",
       "  'I\\'m a big fan of \"Stalker\" and \"Mirror\", John. If you enjoyed \"Battleship Potemkin\" and \"Solaris\", I think you\\'ll love these films too! \"Stalker\" is a sci-fi film that explores themes of humanity and the search for meaning, while \"Mirror\" is a deeply personal and experimental film from one of Russia\\'s most renowned directors. I hope you give them a try! '),\n",
       " (19,\n",
       "  'John',\n",
       "  'I\\'m going to watch \"Battleship Potemkin\" and \"Solaris\" soon. I\\'ll let you know what I think of them! I\\'m also curious about \"Stalker\" and \"Mirror\". Do you have any other Soviet movie recommendations? '),\n",
       " (20,\n",
       "  'Mary',\n",
       "  'John, I\\'d also recommend checking out \"The Cranes Are Flying\" and \"Letter Never Sent\". \"The Cranes Are Flying\" is a heart-wrenching drama that explores the themes of love and loss during World War II, while \"Letter Never Sent\" is a thrilling adventure set in the Siberian wilderness. I think you\\'ll enjoy them! '),\n",
       " (21,\n",
       "  'John',\n",
       "  'I\\'ll have to check out \"The Cranes Are Flying\" and \"Letter Never Sent.\" I\\'m curious to see how they compare to \"Battleship Potemkin\" and \"Solaris.\" Do you have any other Soviet film recommendations? '),\n",
       " (22,\n",
       "  'Mary',\n",
       "  'Hey John, I just remembered two more Soviet films you might enjoy! \"The Cranes Are Flying\" is a heart-wrenching drama set during World War II, and \"Letter Never Sent\" is a thrilling adventure in the Siberian wilderness. I think you\\'ll love them! '),\n",
       " (23,\n",
       "  'John',\n",
       "  'Thanks for the additional film recommendations, Mary! I\\'m looking forward to watching \"The Cranes Are Flying\" and \"Letter Never Sent.\" I\\'ll let you know what I think of them! Do you have any other favorite Soviet films or genres you\\'d like to share? '),\n",
       " (24,\n",
       "  'Mary',\n",
       "  'John, here are two more Soviet films you might enjoy: \"The Cranes Are Flying\" and \"Letter Never Sent.\" \"The Cranes Are Flying\" is a heart-wrenching drama set during World War II, and \"Letter Never Sent\" is a thrilling adventure in the Siberian wilderness. I think you\\'ll love them! '),\n",
       " (25,\n",
       "  'John',\n",
       "  'I can\\'t wait to watch \"The Cranes Are Flying\" and \"Letter Never Sent.\" I\\'ll let you know what I think of them! I\\'m also curious if you have any other favorite Soviet films or genres you\\'d like to share. I\\'m always open to discovering new things! ')]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "chat=ChatThread(agent_list=[John, Mary], neutral_llm=LLMApi())\n",
    "chat.run_chat(max_turns=25)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "# load the last chat\n",
    "import os\n",
    "from chat_llm import load_chat\n",
    "from chat_llm import ChatThread, Agent\n",
    "from advanced_agents import AdvancedAgent\n",
    "# last chat by time\n",
    "# load the last chat\n",
    "\n",
    "# get the list of chat files\n",
    "chat_files = os.listdir(\"chat_logs\")\n",
    "# sort the chat files based on the timestamp\n",
    "sorted_chat_files = sorted(chat_files, key=lambda x: os.path.getmtime(f\"chat_logs/{x}\"), reverse=True)\n",
    "\n",
    "# get the path of the last chat file\n",
    "last_chat_path = sorted_chat_files[0]\n",
    "\n",
    "# load the last chat\n",
    "chat = load_chat(f\"chat_logs/{last_chat_path}\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "ename": "AttributeError",
     "evalue": "'NoneType' object has no attribute 'execute'",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
      "Cell \u001b[0;32mIn[2], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m chat\u001b[38;5;241m.\u001b[39magent_list[\u001b[38;5;241m0\u001b[39m]\u001b[38;5;241m.\u001b[39mmemory\u001b[38;5;241m.\u001b[39msearch(\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mSELECT text FROM txtai WHERE turn=20\u001b[39m\u001b[38;5;124m\"\u001b[39m, limit\u001b[38;5;241m=\u001b[39m\u001b[38;5;241m5\u001b[39m)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/embeddings/base.py:363\u001b[0m, in \u001b[0;36mEmbeddings.search\u001b[0;34m(self, query, limit, weights, index, parameters)\u001b[0m\n\u001b[1;32m    347\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21msearch\u001b[39m(\u001b[38;5;28mself\u001b[39m, query, limit\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, weights\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, index\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, parameters\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m):\n\u001b[1;32m    348\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    349\u001b[0m \u001b[38;5;124;03m    Finds documents most similar to the input query. This method will run either an index search\u001b[39;00m\n\u001b[1;32m    350\u001b[0m \u001b[38;5;124;03m    or an index + database search depending on if a database is available.\u001b[39;00m\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    360\u001b[0m \u001b[38;5;124;03m        list of (id, score) for index search, list of dict for an index + database search\u001b[39;00m\n\u001b[1;32m    361\u001b[0m \u001b[38;5;124;03m    \"\"\"\u001b[39;00m\n\u001b[0;32m--> 363\u001b[0m     results \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mbatchsearch([query], limit, weights, index, [parameters])\n\u001b[1;32m    364\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m results[\u001b[38;5;241m0\u001b[39m] \u001b[38;5;28;01mif\u001b[39;00m results \u001b[38;5;28;01melse\u001b[39;00m results\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/embeddings/base.py:382\u001b[0m, in \u001b[0;36mEmbeddings.batchsearch\u001b[0;34m(self, queries, limit, weights, index, parameters)\u001b[0m\n\u001b[1;32m    366\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mbatchsearch\u001b[39m(\u001b[38;5;28mself\u001b[39m, queries, limit\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, weights\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, index\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m, parameters\u001b[38;5;241m=\u001b[39m\u001b[38;5;28;01mNone\u001b[39;00m):\n\u001b[1;32m    367\u001b[0m \u001b[38;5;250m    \u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    368\u001b[0m \u001b[38;5;124;03m    Finds documents most similar to the input queries. This method will run either an index search\u001b[39;00m\n\u001b[1;32m    369\u001b[0m \u001b[38;5;124;03m    or an index + database search depending on if a database is available.\u001b[39;00m\n\u001b[0;32m   (...)\u001b[0m\n\u001b[1;32m    379\u001b[0m \u001b[38;5;124;03m        list of (id, score) per query for index search, list of dict per query for an index + database search\u001b[39;00m\n\u001b[1;32m    380\u001b[0m \u001b[38;5;124;03m    \"\"\"\u001b[39;00m\n\u001b[0;32m--> 382\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m Search(\u001b[38;5;28mself\u001b[39m)(queries, limit, weights, index, parameters)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/embeddings/search/base.py:70\u001b[0m, in \u001b[0;36mSearch.__call__\u001b[0;34m(self, queries, limit, weights, index, parameters)\u001b[0m\n\u001b[1;32m     68\u001b[0m \u001b[38;5;66;03m# Database search\u001b[39;00m\n\u001b[1;32m     69\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mindexids \u001b[38;5;129;01mand\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mdatabase:\n\u001b[0;32m---> 70\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mdbsearch(queries, limit, weights, index, parameters)\n\u001b[1;32m     72\u001b[0m \u001b[38;5;66;03m# Default vector index query (sparse, dense or hybrid)\u001b[39;00m\n\u001b[1;32m     73\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39msearch(queries, limit, weights, index)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/embeddings/search/base.py:241\u001b[0m, in \u001b[0;36mSearch.dbsearch\u001b[0;34m(self, queries, limit, weights, index, parameters)\u001b[0m\n\u001b[1;32m    238\u001b[0m results \u001b[38;5;241m=\u001b[39m []\n\u001b[1;32m    239\u001b[0m \u001b[38;5;28;01mfor\u001b[39;00m x, query \u001b[38;5;129;01min\u001b[39;00m \u001b[38;5;28menumerate\u001b[39m(queries):\n\u001b[1;32m    240\u001b[0m     \u001b[38;5;66;03m# Run the database query, get matching bulk searches for current query\u001b[39;00m\n\u001b[0;32m--> 241\u001b[0m     result \u001b[38;5;241m=\u001b[39m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mdatabase\u001b[38;5;241m.\u001b[39msearch(query, [r \u001b[38;5;28;01mfor\u001b[39;00m y, r \u001b[38;5;129;01min\u001b[39;00m scan \u001b[38;5;28;01mif\u001b[39;00m x \u001b[38;5;241m==\u001b[39m y], limit, parameters[x] \u001b[38;5;28;01mif\u001b[39;00m parameters \u001b[38;5;129;01mand\u001b[39;00m parameters[x] \u001b[38;5;28;01melse\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m)\n\u001b[1;32m    242\u001b[0m     results\u001b[38;5;241m.\u001b[39mappend(result)\n\u001b[1;32m    244\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m results\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/database/base.py:163\u001b[0m, in \u001b[0;36mDatabase.search\u001b[0;34m(self, query, similarity, limit, parameters)\u001b[0m\n\u001b[1;32m    160\u001b[0m query[\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mwhere\u001b[39m\u001b[38;5;124m\"\u001b[39m] \u001b[38;5;241m=\u001b[39m where\n\u001b[1;32m    162\u001b[0m \u001b[38;5;66;03m# Run query\u001b[39;00m\n\u001b[0;32m--> 163\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mquery(query, limit, parameters)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/database/rdbms.py:214\u001b[0m, in \u001b[0;36mRDBMS.query\u001b[0;34m(self, query, limit, parameters)\u001b[0m\n\u001b[1;32m    212\u001b[0m \u001b[38;5;66;03m# Clear scores when no similar clauses present\u001b[39;00m\n\u001b[1;32m    213\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m similarity:\n\u001b[0;32m--> 214\u001b[0m     \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mscores(\u001b[38;5;28;01mNone\u001b[39;00m)\n\u001b[1;32m    216\u001b[0m \u001b[38;5;66;03m# Runs a user query through execute method, which has common user query handling logic\u001b[39;00m\n\u001b[1;32m    217\u001b[0m args \u001b[38;5;241m=\u001b[39m (query, parameters) \u001b[38;5;28;01mif\u001b[39;00m parameters \u001b[38;5;28;01melse\u001b[39;00m (query,)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/database/rdbms.py:447\u001b[0m, in \u001b[0;36mRDBMS.scores\u001b[0;34m(self, similarity)\u001b[0m\n\u001b[1;32m    439\u001b[0m \u001b[38;5;250m\u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    440\u001b[0m \u001b[38;5;124;03mLoads a batch of similarity scores to a temporary table for efficient query processing.\u001b[39;00m\n\u001b[1;32m    441\u001b[0m \n\u001b[1;32m    442\u001b[0m \u001b[38;5;124;03mArgs:\u001b[39;00m\n\u001b[1;32m    443\u001b[0m \u001b[38;5;124;03m    similarity: similarity results as [(indexid, score)]\u001b[39;00m\n\u001b[1;32m    444\u001b[0m \u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    446\u001b[0m \u001b[38;5;66;03m# Create or replace scores table\u001b[39;00m\n\u001b[0;32m--> 447\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mcreatescores()\n\u001b[1;32m    449\u001b[0m \u001b[38;5;66;03m# Delete scores\u001b[39;00m\n\u001b[1;32m    450\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mcursor\u001b[38;5;241m.\u001b[39mexecute(Statement\u001b[38;5;241m.\u001b[39mDELETE_SCORES)\n",
      "File \u001b[0;32m~/anaconda3/lib/python3.11/site-packages/txtai/database/rdbms.py:470\u001b[0m, in \u001b[0;36mRDBMS.createscores\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    465\u001b[0m \u001b[38;5;250m\u001b[39m\u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    466\u001b[0m \u001b[38;5;124;03mCreates temporary scores table.\u001b[39;00m\n\u001b[1;32m    467\u001b[0m \u001b[38;5;124;03m\"\"\"\u001b[39;00m\n\u001b[1;32m    469\u001b[0m \u001b[38;5;66;03m# Create or Replace temporary scores table\u001b[39;00m\n\u001b[0;32m--> 470\u001b[0m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mcursor\u001b[38;5;241m.\u001b[39mexecute(Statement\u001b[38;5;241m.\u001b[39mCREATE_SCORES)\n",
      "\u001b[0;31mAttributeError\u001b[0m: 'NoneType' object has no attribute 'execute'"
     ]
    }
   ],
   "source": [
    "chat.agent_list[0].memory.search(\"SELECT text FROM txtai WHERE turn=20\", limit=5)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}