Update agentic.py

agentic.py

@@ -473,7 +473,7 @@ Now provide your response immediately without any preamble.
             "data": image_base64,
         }
 
-        vision_provider = "mistral"
+        vision_provider = os.getenv("VLM_PROVIDER", "mistral")
 
         if vision_provider == "openai":
             image_handling = openai_image_handling
@@ -762,7 +762,7 @@ Now provide your response immediately respecting this format: 'next node: \'the
     """
 
     downloaded = ""
-    # If there's an input file, download it directly
+    # If there's an input file, download it directlynext node:
     if state.get("input_file", None):
         downloaded = download_input_file(state.get("task_id"))
 
@@ -772,7 +772,7 @@ Now provide your response immediately respecting this format: 'next node: \'the
 
     entry_node_response[-1].pretty_print()
 
-    regex_result = re.search(r'.*next\snode:\s+(?P<next_node>.*)$', entry_node_response[-1].content)
+    regex_result = re.search(r'.*next\snode:\s+(?P<next_node>[a-z_]*)$', entry_node_response[-1].content)
 
     next_node = "END"
     if regex_result:
@@ -837,10 +837,11 @@ if __name__ == "__main__":
     
     # print(vision_node.__doc__)
 
-    input = {
-        "task_id": "9d191bce-651d-4746-be2d-7ef8ecadb9c2",
-        "question": "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec. What does Teal'c say in response to the question \"Isn't that hot?\"", 
-        "file_name": ""
+    input =  {
+        "task_id": "cca530fc-4052-43b2-b130-b30968d8aa44",
+        "question": "Review the chess position provided in the image. It is black's turn. Provide the correct next move for black which guarantees a win. Please provide your response in algebraic notation.",
+        "Level": "1",
+        "file_name": "cca530fc-4052-43b2-b130-b30968d8aa44.png"
     }
 
     question = input.get("question", "No question found")