Skip to content

Commit 0a6ebd2

Browse files
[FEATURE] Added function to get model name of LLM (#95)
* Added function to get model name of LLM * Version bump * Version bump
1 parent 7613bec commit 0a6ebd2

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

src/unstract/sdk/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.46.0"
1+
__version__ = "0.47.0"
22

33

44
def get_sdk_version():

src/unstract/sdk/llm.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,17 @@ def get_class_name(self) -> str:
188188
"""
189189
return self._llm_instance.class_name()
190190

191+
def get_model_name(self) -> str:
192+
"""Gets the name of the LLM model
193+
194+
Args:
195+
NA
196+
197+
Returns:
198+
LLM model name
199+
"""
200+
return self._llm_instance.model
201+
191202
@deprecated("Use LLM instead of ToolLLM")
192203
def get_llm(self, adapter_instance_id: Optional[str] = None) -> LlamaIndexLLM:
193204
if not self._llm_instance:

0 commit comments

Comments
 (0)