File size: 1,377 Bytes
78683dc
 
 
 
 
 
 
 
 
 
d8b8502
7828173
9323159
90c6fb1
 
3716a9b
90c6fb1
 
 
7828173
90c6fb1
6470a23
 
36dac61
 
 
 
8f8fb45
6470a23
7cb9551
6470a23
133de3a
90c6fb1
39ec4fe
133de3a
6470a23
7828173
36dac61
0e701d6
7828173
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import smolagents, numpy, math, xlrd, os
from smolagents import (
    CodeAgent,
    HfApiModel,
    InferenceClientModel,
    WebSearchTool,
    PythonInterpreterTool,
    FinalAnswerTool,
    DuckDuckGoSearchTool,
    GoogleSearchTool,
    )

class newAgent:
    """Adapts smolagents.CodeAgent to the HF course template API."""
    def __init__(self):
        model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"  # correct repo name
        hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")   # read real secret
        if not hf_token:
            raise RuntimeError("HUGGINGFACEHUB_API_TOKEN not set in Space secrets")

        model = HfApiModel(model_id=model_id, token=hf_token)
        
        # include FinalAnswerTool in tools so agent knows when to stop
        tools = [FinalAnswerTool()]
        self.agent = CodeAgent(
            tools=tools,
            model=model,
            add_base_tools=True,
            max_steps=5 # limit reasoning time
        )
   
    def __call__(self, question: str) -> str:
        """ONE question in β†’ ONE pure-text answer out."""
        #↓ Replace .run with whatever method actually returns the answer string.
        result = self.agent.run(question)
        return result

        #answer = self.run
          
#agent.run(
#    "At what temperature and for how long should I bake French baguettes made with type 65 flour?",
#)