71 lines
2.6 KiB
Python
71 lines
2.6 KiB
Python
import time
|
|
import random
|
|
from typing import Any, Callable
|
|
from src import ai_client
|
|
|
|
class UserSimAgent:
|
|
def __init__(self, hook_client: Any, model: str = "gemini-2.5-flash-lite", enable_delays: bool = True) -> None:
|
|
self.hook_client = hook_client
|
|
self.model = model
|
|
self.enable_delays = enable_delays
|
|
self.system_prompt = (
|
|
"You are a software engineer testing an AI coding assistant called 'Manual Slop'. "
|
|
"You want to build a small Python project and verify the assistant's capabilities. "
|
|
"Keep your responses concise and human-like. "
|
|
"Do not use markdown blocks for your main message unless you are providing code."
|
|
)
|
|
|
|
def calculate_reading_delay(self, text: str) -> float:
|
|
word_count = len(text.split())
|
|
delay = (word_count / 200.0) * 60.0 * 0.5
|
|
return max(0.5, min(5.0, delay))
|
|
|
|
def wait_to_read(self, text: str) -> None:
|
|
if self.enable_delays:
|
|
delay = self.calculate_reading_delay(text)
|
|
time.sleep(delay)
|
|
|
|
def wait_to_think(self, probability: float = 0.2, min_delay: float = 2.0, max_delay: float = 5.0) -> None:
|
|
if self.enable_delays and random.random() < probability:
|
|
delay = random.uniform(min_delay, max_delay)
|
|
time.sleep(delay)
|
|
|
|
def simulate_typing(self, text: str, jitter_range: tuple[float, float] = (0.01, 0.05)) -> None:
|
|
if self.enable_delays:
|
|
# Simulate typing by sleeping after chunks or characters to balance speed and realism
|
|
if len(text) > 200:
|
|
for i in range(0, len(text), 10):
|
|
time.sleep(random.uniform(jitter_range[0] * 3, jitter_range[1] * 3))
|
|
elif len(text) > 50:
|
|
for i in range(0, len(text), 3):
|
|
time.sleep(random.uniform(jitter_range[0] * 1.5, jitter_range[1] * 1.5))
|
|
else:
|
|
for char in text:
|
|
time.sleep(random.uniform(jitter_range[0], jitter_range[1]))
|
|
|
|
def generate_response(self, conversation_history: list[dict]) -> str:
|
|
"""
|
|
Generates a human-like response based on the conversation history.
|
|
conversation_history: list of dicts with 'role' and 'content'
|
|
"""
|
|
last_ai_msg = ""
|
|
for entry in reversed(conversation_history):
|
|
if entry.get('role') == 'AI':
|
|
last_ai_msg = entry.get('content', '')
|
|
break
|
|
try:
|
|
ai_client.set_custom_system_prompt(self.system_prompt)
|
|
response = ai_client.send(md_content="", user_message=last_ai_msg)
|
|
finally:
|
|
ai_client.set_custom_system_prompt("")
|
|
return response
|
|
|
|
def perform_action_with_delay(self, action_func: Callable, *args: Any, **kwargs: Any) -> Any:
|
|
"""
|
|
Executes an action with a human-like delay if enabled.
|
|
"""
|
|
if self.enable_delays:
|
|
delay = random.uniform(0.5, 2.0)
|
|
time.sleep(delay)
|
|
return action_func(*args, **kwargs)
|