Update src/agent.py

src/agent.py

@@ -95,8 +95,11 @@ class BasicAgent():
         retry_count = 0
 
         question = f"Question: {question}"
-        context = f"Story content:\n{file_content}"
-        print(f"Starting Agent with question text: {question}, {file_content}")  
+        if file_content:
+            context = f"Story content:\n{file_content}"
+        elif file_path:
+            context = f"File path: {file_path}"
+        print(f"Starting Agent with question text: {question}, {context}")  
         while True:
             try:
                 # Run the agent