Chamin09 commited on
Commit
09e0044
·
verified ·
1 Parent(s): 122f517

imported `sentence_transformer` to CoordinatorAgent

Browse files
Files changed (1) hide show
  1. src/agents/coordinator_agent.py +1 -0
src/agents/coordinator_agent.py CHANGED
@@ -2,6 +2,7 @@ from typing import Dict, Any, Optional
2
  from .base_agent import BaseAgent
3
  import numpy as np
4
  from transformers import AutoTokenizer, AutoModel
 
5
  import torch
6
 
7
  class CoordinatorAgent(BaseAgent):
 
2
  from .base_agent import BaseAgent
3
  import numpy as np
4
  from transformers import AutoTokenizer, AutoModel
5
+ from sentence_transformers import SentenceTransformer
6
  import torch
7
 
8
  class CoordinatorAgent(BaseAgent):