mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 02:34:29 +00:00
41e5b3aa8d
* feat(humanloop.py): initial commit for humanloop prompt management integration Closes https://github.com/BerriAI/litellm/issues/213 * feat(humanloop.py): working e2e humanloop prompt management integration Closes https://github.com/BerriAI/litellm/issues/213 * fix(humanloop.py): fix linting errors * fix: fix linting erro * fix: fix test * test: handle filenotfound error
35 lines
1 KiB
Python
35 lines
1 KiB
Python
import os
|
|
import sys
|
|
import threading
|
|
from datetime import datetime
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system-path
|
|
|
|
import pytest
|
|
from litellm.integrations.humanloop import HumanLoopPromptManager
|
|
from litellm.types.utils import StandardCallbackDynamicParams
|
|
from litellm.litellm_core_utils.litellm_logging import DynamicLoggingCache
|
|
from unittest.mock import Mock, patch
|
|
|
|
|
|
def test_compile_prompt():
|
|
prompt_manager = HumanLoopPromptManager()
|
|
prompt_template = [
|
|
{
|
|
"content": "You are {{person}}. Answer questions as this person. Do not break character.",
|
|
"name": None,
|
|
"tool_call_id": None,
|
|
"role": "system",
|
|
"tool_calls": None,
|
|
}
|
|
]
|
|
prompt_variables = {"person": "John"}
|
|
compiled_prompt = prompt_manager._compile_prompt_helper(
|
|
prompt_template, prompt_variables
|
|
)
|
|
assert (
|
|
compiled_prompt[0]["content"]
|
|
== "You are John. Answer questions as this person. Do not break character."
|
|
)
|