Saptadip Sarkar commited on
Commit
e890401
·
1 Parent(s): 5164dc5

fix: huggingface llm package name

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import requests
4
  import inspect
5
  import pandas as pd
6
  from llama_index.core.agent.workflow import AgentWorkflow
7
- from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
8
  #from llama_index.core.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
9
  import agent_course_tools as tools
10
 
@@ -17,7 +17,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
17
  class BasicAgent:
18
  def __init__(self):
19
  print("BasicAgent initialized.")
20
- llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
21
  agent = AgentWorkflow.from_tools_or_functions([
22
  tools.wiki_tool,
23
  tools.ddgo_tool
 
4
  import inspect
5
  import pandas as pd
6
  from llama_index.core.agent.workflow import AgentWorkflow
7
+ from llama_index.llms.huggingface import HuggingFaceLLM
8
  #from llama_index.core.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
9
  import agent_course_tools as tools
10
 
 
17
  class BasicAgent:
18
  def __init__(self):
19
  print("BasicAgent initialized.")
20
+ llm = HuggingFaceLLM(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
21
  agent = AgentWorkflow.from_tools_or_functions([
22
  tools.wiki_tool,
23
  tools.ddgo_tool
requirements.txt CHANGED
@@ -4,4 +4,5 @@ llama-index
4
  llama-index-tools-wikipedia
5
  llama-index-llms-litellm
6
  llama-index-readers-file
7
- llama-index-tools-duckduckgo
 
 
4
  llama-index-tools-wikipedia
5
  llama-index-llms-litellm
6
  llama-index-readers-file
7
+ llama-index-tools-duckduckgo
8
+ llama-index-llms-huggingface