feat: add llm keywords to language model base (#8115)

* update the keywords

* Update test_chatollama_component.py

* [autofix.ci] apply automated fixes

---------

Co-authored-by: Yuqi Tang <yuqi.tang@datastax.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Edwin Jose 2025-05-23 16:00:02 -04:00 committed by GitHub
commit 4343839dc6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
24 changed files with 249 additions and 31 deletions

View file

@ -26,6 +26,14 @@ class LCModelComponent(Component):
display_name: str = "Model Name"
description: str = "Model Description"
trace_type = "llm"
metadata = {
"keywords": [
"model",
"llm",
"language model",
"large language model",
],
}
# Optional output parser to pass to the runnable. Subclasses may allow the user to input an `output_parser`
output_parser: BaseOutputParser | None = None

View file

@ -1281,7 +1281,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -1665,7 +1672,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -2049,7 +2063,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -902,7 +902,14 @@
"frozen": false,
"icon": "OpenAI",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -784,7 +784,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -1399,7 +1399,14 @@
"icon": "Anthropic",
"key": "AnthropicModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -559,7 +559,14 @@
"icon": "Novita",
"legacy": false,
"lf_version": "1.1.4.post1",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -1161,7 +1161,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -1959,7 +1959,14 @@
"icon": "SambaNova",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -150,7 +150,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -885,7 +885,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.3.2",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -977,7 +977,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -2778,7 +2778,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -3162,7 +3169,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -2287,7 +2287,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -463,7 +463,14 @@
"icon": "OpenAI",
"legacy": false,
"lf_version": "1.1.5",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -1599,7 +1606,14 @@
"icon": "OpenAI",
"legacy": false,
"lf_version": "1.1.5",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -1235,7 +1235,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -299,7 +299,14 @@
"key": "AnthropicModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -614,7 +621,14 @@
"key": "AnthropicModel",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -2354,7 +2354,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -2738,7 +2745,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -397,7 +397,14 @@
"key": "AnthropicModel",
"legacy": false,
"lf_version": "1.1.5",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -889,7 +889,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -805,7 +805,14 @@
"icon": "OpenAI",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -1176,7 +1183,14 @@
"icon": "OpenAI",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [
@ -1667,7 +1681,14 @@
"icon": "OpenAI",
"legacy": false,
"lf_version": "1.2.0",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -1823,7 +1823,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -2620,7 +2620,14 @@
"icon": "OpenAI",
"key": "OpenAIModel",
"legacy": false,
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -761,7 +761,14 @@
"key": "OpenAIModel",
"legacy": false,
"lf_version": "1.1.3",
"metadata": {},
"metadata": {
"keywords": [
"model",
"llm",
"language model",
"large language model"
]
},
"minimized": false,
"output_types": [],
"outputs": [

View file

@ -51,7 +51,7 @@ class TestChatOllamaComponent(ComponentTestBaseWithoutClient):
model="ollama-model",
mirostat=0,
format="json",
metadata={},
metadata={"keywords": ["model", "llm", "language model", "large language model"]},
num_ctx=2048,
num_gpu=1,
num_thread=4,