Rahul-8799 commited on
Commit
cc2b268
·
verified ·
1 Parent(s): 763d31e

Update agents/project_manager_agent.py

Browse files
Files changed (1) hide show
  1. agents/project_manager_agent.py +1 -0
agents/project_manager_agent.py CHANGED
@@ -1,5 +1,6 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
 
3
 
4
  MODEL_REPO = "Rahul-8799/project_manager_gemma3"
5
 
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
+ from langchain_core.messages import AIMessage
4
 
5
  MODEL_REPO = "Rahul-8799/project_manager_gemma3"
6