Update app.py

app.py

@@ -37,8 +37,7 @@ def write_article(topic: str, search_results: str) -> str:
         작성된 기사
     """
     try:
-        system_prompt = """
-        당신은 전문 기자입니다. 주어진 주제에 대해 검색된 정보를 바탕으로 새로운 기사를 한국어로 작성해야 합니다.
+        system_prompt = """당신은 전문 기자입니다. 주어진 주제에 대해 검색된 정보를 바탕으로 새로운 기사를 한국어로 작성해야 합니다.
         다음 형식으로 기사를 작성해주세요:
 
         1. 제목 (흥미롭고 눈에 띄는 제목)
@@ -46,20 +45,29 @@ def write_article(topic: str, search_results: str) -> str:
         3. 본문 (상세한 내용을 단락으로 구분하여 작성)
         4. 결론 (기사의 의의나 향후 전망)
 
-        기사는 객관적이고 사실에 기반하여 작성해야 하며, 검색된 정보를 재구성하여 새로운 시각으로 작성해주세요.
-        """
+        기사는 객관적이고 사실에 기반하여 작성해야 하며, 검색된 정보를 재구성하여 새로운 시각으로 작성해주세요."""
         
-        prompt = f"""
-        다음 주제와 검색 결과를 바탕으로 새로운 기사를 한국어로 작성해주세요:
+        user_prompt = f"""다음 주제와 검색 결과를 바탕으로 새로운 기사를 한국어로 작성해주세요:
 
         주제: {topic}
 
         검색 결과:
         {search_results}
 
-        위 정보를 바탕으로 새롭고 통찰력 있는 기사를 5문단으로 작성해주세요.
-        """
-        return model.generate(system_prompt + prompt)
+        위 정보를 바탕으로 새롭고 통찰력 있는 기사를 작성해주세요."""
+
+        messages = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt}
+        ]
+        
+        response = model(messages)
+        if isinstance(response, dict) and 'content' in response:
+            return response['content']
+        elif hasattr(response, 'content'):
+            return response.content
+        else:
+            return str(response)
     except Exception as e:
         traceback.print_exc()
         return f"기사 작성 중 오류 발생: {str(e)}"
@@ -79,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[search_news, write_article, final_answer],
-    max_steps=6,
+    max_steps=10,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,