Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions promptmodel/chat_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,6 +107,26 @@ def get_config(self) -> ChatModelConfig:
)
return ChatModelConfig(prompt, version_detail, message_logs)

def get_prompts(self) -> str:
"""Get prompts for the ChatModel. (Deprecated)
Returns:
str: system prompt for the ChatModel.
"""
prompt, version_detail, message_logs = run_async_in_sync(
LLMProxy.fetch_chat_model(self.name, self.session_uuid, self.version)
)
return prompt

def get_messages(self) -> List[Dict[str, Any]]:
"""Get message logs for the ChatModel. (Deprecated)
Returns:
List[Dict[str, Any]]: messages for the ChatModel.
"""
prompt, version_detail, message_logs = run_async_in_sync(
LLMProxy.fetch_chat_model(self.name, self.session_uuid, self.version)
)
return message_logs

@check_connection_status_decorator
def add_messages(
self,
Expand Down
13 changes: 13 additions & 0 deletions promptmodel/function_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,19 @@ def get_config(self, *args, **kwargs) -> FunctionModelConfig:
)
return FunctionModelConfig(prompt, version_detail)

@check_connection_status_decorator
def get_prompts(self, *args, **kwargs) -> List[Dict[str, Any]]:
"""Get prompts for the promptmodel. (Deprecated)
Returns:
List[Dict]: prompts
"""
# add name to the list of function_models

prompt, version_detail = run_async_in_sync(
LLMProxy.fetch_prompts(self.name, self.version)
)
return prompt

@check_connection_status_decorator
def run(
self,
Expand Down