Niki Zhang commited on
Commit
6549464
·
verified ·
1 Parent(s): 433556e

Update chatbox.py

Browse files
Files changed (1) hide show
  1. chatbox.py +6 -6
chatbox.py CHANGED
@@ -11,7 +11,7 @@ import argparse
11
  import inspect
12
 
13
  from langchain.agents.initialize import initialize_agent
14
- from langchain.agents.tools import Tool
15
  from langchain.chains.conversation.memory import ConversationBufferMemory
16
  from langchain_community.chat_models import ChatOpenAI
17
  import torch
@@ -130,11 +130,11 @@ def build_chatbot_tools(load_dict):
130
  **{name: models[name] for name in template_required_names})
131
 
132
  tools = []
133
- for instance in models.values():
134
- for e in dir(instance):
135
- if e.startswith('inference'):
136
- func = getattr(instance, e)
137
- tools.append(Tool(name=func.name, description=func.description, func=func))
138
  return tools
139
 
140
  class ConversationBot:
 
11
  import inspect
12
 
13
  from langchain.agents.initialize import initialize_agent
14
+ # from langchain.agents.tools import Tool
15
  from langchain.chains.conversation.memory import ConversationBufferMemory
16
  from langchain_community.chat_models import ChatOpenAI
17
  import torch
 
130
  **{name: models[name] for name in template_required_names})
131
 
132
  tools = []
133
+ # for instance in models.values():
134
+ # for e in dir(instance):
135
+ # if e.startswith('inference'):
136
+ # func = getattr(instance, e)
137
+ # tools.append(Tool(name=func.name, description=func.description, func=func))
138
  return tools
139
 
140
  class ConversationBot: