Fixed a bug in window update

Author: yeshaokai

amadeusgpt/agentic/video_llm.py

@@ -137,7 +137,7 @@ def speak(self, video_data):
         self.update_history("system", self.system_prompt)
         self.update_history(
             "user", "This video is about a mouse in its home cage.", multi_image_content=multi_image_content, in_place = True)
-
+         
         response = self.connect_gpt(self.context_window)
         text = response.choices[0].message.content.strip()
         print(text)
@@ -153,7 +153,6 @@ def speak(self, video_data):
     video_sampler = VideoSampler(video_path, segment_duration, frames_per_segment)
     video_data = video_sampler.process_video()
 
-    video_data = {k:v for k,v in video_data.items() if k < 1}
-    print (video_data)
+    video_data = {k:v for k,v in video_data.items() if k < 30}
     video_llm = VideoLLM(config=config)
     video_llm.speak(video_data)

amadeusgpt/analysis_objects/llm.py

@@ -20,7 +20,7 @@ class LLM(AnalysisObject):
 
     def __init__(self, config):
         self.config = config
-        self.max_tokens = config.get("max_tokens", 36000)
+        self.max_tokens = config.get("max_tokens", 4096)
         self.gpt_model = config.get("gpt_model", "gpt-4o")
         self.keep_last_n_messages = config.get("keep_last_n_messages", 2)
 
@@ -168,11 +168,13 @@ def update_history(self, role, content, multi_image_content = None, in_place=Fal
                 self.context_window.pop(1)
 
             if in_place == True:
-                assert len(self.context_window) <= 2, "context window should have only 2 elements"
+                assert len(self.context_window) <= 2, "context window should have no more than 2 elements"
                 if len(self.context_window) == 2:
                     self.context_window[1] = new_message
                 else:
                     self.context_window.append(new_message)
+            else:
+                self.context_window.append(new_message)
 
 
     def clean_context_window(self):