ahnhs2k commited on
Commit
abcc550
ยท
1 Parent(s): a85b201
Files changed (1) hide show
  1. agent.py +9 -5
agent.py CHANGED
@@ -828,10 +828,11 @@ class BasicAgent:
828
  # ๊ทธ๋ž˜ํ”„๋Š” ๋ชจ๋“ˆ ๋กœ๋“œ ์‹œ ์ปดํŒŒ์ผ๋จ. ์—ฌ๊ธฐ์„œ๋Š” ์ƒํƒœ๋งŒ ์•Œ๋ฆผ.
829
  print("โœ… GAIA Agent initialized (LangGraph StateGraph)")
830
 
831
- def __call__(self, question: str) -> str:
832
  """
833
- app.py์—์„œ question_text๋งŒ ๋ฐ›๋Š” ๊ณ ์ • ์ธํ„ฐํŽ˜์ด์Šค.
834
- - recursion_limit์€ invoke config์—์„œ ์ œํ•œ (LangGraph ๋ฒ„์ „ ํ˜ธํ™˜)
 
835
  """
836
  state: AgentState = {
837
  "question": question,
@@ -842,6 +843,9 @@ class BasicAgent:
842
  "steps": 0,
843
  }
844
 
845
- # LangGraph ๋ฌดํ•œ๋ฃจํ”„/์žฌ๊ท€ ๋ณดํ˜ธ
846
- out = GRAPH.invoke(state, config={"recursion_limit": 12})
 
 
 
847
  return clean_final_answer(out.get("answer", ""))
 
828
  # ๊ทธ๋ž˜ํ”„๋Š” ๋ชจ๋“ˆ ๋กœ๋“œ ์‹œ ์ปดํŒŒ์ผ๋จ. ์—ฌ๊ธฐ์„œ๋Š” ์ƒํƒœ๋งŒ ์•Œ๋ฆผ.
829
  print("โœ… GAIA Agent initialized (LangGraph StateGraph)")
830
 
831
+ def __call__(self, question: str, **kwargs) -> str:
832
  """
833
+ app.py / Gradio / HF OAuth ๋ž˜ํผ๊ฐ€
834
+ task_id ๊ฐ™์€ keyword argument๋ฅผ ๋„˜๊ฒจ๋„
835
+ ๋ฌด์กฐ๊ฑด ๋ฌด์‹œํ•˜๊ณ  question๋งŒ ์ฒ˜๋ฆฌํ•œ๋‹ค.
836
  """
837
  state: AgentState = {
838
  "question": question,
 
843
  "steps": 0,
844
  }
845
 
846
+ out = GRAPH.invoke(
847
+ state,
848
+ config={"recursion_limit": 12}
849
+ )
850
+
851
  return clean_final_answer(out.get("answer", ""))