Kevin Hu
commited on
Commit
·
ef2346f
1
Parent(s):
53b1865
Add more mistral models. (#3826)
Browse files### What problem does this PR solve?
#3647
### Type of change
- [x] New Feature (non-breaking change which adds functionality)
- agent/canvas.py +1 -0
- conf/llm_factories.json +34 -10
agent/canvas.py
CHANGED
@@ -234,6 +234,7 @@ class Canvas(ABC):
|
|
234 |
except Exception as e:
|
235 |
yield {"content": "*Exception*: {}".format(e), "running_status": True}
|
236 |
logging.exception("Canvas.run got exception")
|
|
|
237 |
|
238 |
if ran >= len(self.path[-1]) and waiting:
|
239 |
without_dependent_checking = waiting
|
|
|
234 |
except Exception as e:
|
235 |
yield {"content": "*Exception*: {}".format(e), "running_status": True}
|
236 |
logging.exception("Canvas.run got exception")
|
237 |
+
ran += 1
|
238 |
|
239 |
if ran >= len(self.path[-1]) and waiting:
|
240 |
without_dependent_checking = waiting
|
conf/llm_factories.json
CHANGED
@@ -618,32 +618,44 @@
|
|
618 |
},
|
619 |
{
|
620 |
"llm_name": "open-mistral-7b",
|
621 |
-
"tags": "LLM,CHAT,
|
622 |
-
"max_tokens":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
623 |
"model_type": "chat"
|
624 |
},
|
625 |
{
|
626 |
"llm_name": "mistral-large-latest",
|
627 |
-
"tags": "LLM,CHAT,
|
628 |
-
"max_tokens":
|
629 |
"model_type": "chat"
|
630 |
},
|
631 |
{
|
632 |
"llm_name": "mistral-small-latest",
|
633 |
-
"tags": "LLM,CHAT,
|
634 |
-
"max_tokens":
|
635 |
"model_type": "chat"
|
636 |
},
|
637 |
{
|
638 |
-
"llm_name": "
|
639 |
"tags": "LLM,CHAT,32k",
|
640 |
"max_tokens": 32000,
|
641 |
"model_type": "chat"
|
642 |
},
|
643 |
{
|
644 |
-
"llm_name": "
|
645 |
-
"tags": "LLM,CHAT,
|
646 |
-
"max_tokens":
|
647 |
"model_type": "chat"
|
648 |
},
|
649 |
{
|
@@ -651,6 +663,18 @@
|
|
651 |
"tags": "LLM,CHAT,8k",
|
652 |
"max_tokens": 8192,
|
653 |
"model_type": "embedding"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
654 |
}
|
655 |
]
|
656 |
},
|
|
|
618 |
},
|
619 |
{
|
620 |
"llm_name": "open-mistral-7b",
|
621 |
+
"tags": "LLM,CHAT,128k",
|
622 |
+
"max_tokens": 128000,
|
623 |
+
"model_type": "chat"
|
624 |
+
},
|
625 |
+
{
|
626 |
+
"llm_name": "ministral-8b-latest",
|
627 |
+
"tags": "LLM,CHAT,128k",
|
628 |
+
"max_tokens": 128000,
|
629 |
+
"model_type": "chat"
|
630 |
+
},
|
631 |
+
{
|
632 |
+
"llm_name": "ministral-3b-latest",
|
633 |
+
"tags": "LLM,CHAT,128k",
|
634 |
+
"max_tokens": 128000,
|
635 |
"model_type": "chat"
|
636 |
},
|
637 |
{
|
638 |
"llm_name": "mistral-large-latest",
|
639 |
+
"tags": "LLM,CHAT,128k",
|
640 |
+
"max_tokens": 128000,
|
641 |
"model_type": "chat"
|
642 |
},
|
643 |
{
|
644 |
"llm_name": "mistral-small-latest",
|
645 |
+
"tags": "LLM,CHAT,128k",
|
646 |
+
"max_tokens": 128000,
|
647 |
"model_type": "chat"
|
648 |
},
|
649 |
{
|
650 |
+
"llm_name": "codestral-latest",
|
651 |
"tags": "LLM,CHAT,32k",
|
652 |
"max_tokens": 32000,
|
653 |
"model_type": "chat"
|
654 |
},
|
655 |
{
|
656 |
+
"llm_name": "mistral-nemo",
|
657 |
+
"tags": "LLM,CHAT,128k",
|
658 |
+
"max_tokens": 128000,
|
659 |
"model_type": "chat"
|
660 |
},
|
661 |
{
|
|
|
663 |
"tags": "LLM,CHAT,8k",
|
664 |
"max_tokens": 8192,
|
665 |
"model_type": "embedding"
|
666 |
+
},
|
667 |
+
{
|
668 |
+
"llm_name": "pixtral-large-latest",
|
669 |
+
"tags": "LLM,CHAT,32k",
|
670 |
+
"max_tokens": 32000,
|
671 |
+
"model_type": "image2text"
|
672 |
+
},
|
673 |
+
{
|
674 |
+
"llm_name": "pixtral-12b",
|
675 |
+
"tags": "LLM,CHAT,32k",
|
676 |
+
"max_tokens": 32000,
|
677 |
+
"model_type": "image2text"
|
678 |
}
|
679 |
]
|
680 |
},
|