Hugging Face's logo

Spaces:
Rahul-8799
/
MAC
Runtime error

App Files Community
Rahul-8799 commited on
Commit
40f1554
·
verified ·
1 Parent(s): b7821ac

Update agents/software_engineer_agent.py

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