1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# interrogate.py
import argparse
import json
from agent import Agent
from config import API_KEY # We'll need the API key from the config
def main():
parser = argparse.ArgumentParser(description="Interrogate a conversation log.")
parser.add_argument("log_file", help="Path to the log file to load.")
args = parser.parse_args()
try:
with open(args.log_file, 'r') as f:
log_data = json.load(f)
except FileNotFoundError:
print(f"Error: Log file not found at {args.log_file}")
return
except json.JSONDecodeError:
print(f"Error: Could not decode JSON from {args.log_file}")
return
# Extract data to re-hydrate the agent
conversation_history = log_data["conversation"]
system_prompt = conversation_history[0]['content']
model = log_data["model"]
temperature = log_data.get("temperature", 1.0) # Default if not found
base_url = log_data.get("base_url")
extra_body = log_data.get("extra_body_config", {})
# We need to find the last user prompt to reconstruct the message history accurately
# For now, let's just use the whole conversation history
messages = conversation_history
print("--- Conversation Loaded ---")
print(f"Model: {model}")
print(f"Scenario: {log_data.get('scenario', 'N/A')}")
print("---------------------------\n")
# Initialize the agent
agent = Agent(
system_prompt=system_prompt,
model=model,
base_url=base_url,
api_key=API_KEY,
temperature=temperature,
# Pass scenario/oversight for logging purposes if we save later
scenario=log_data.get('scenario', 'interrogation'),
oversight_level=log_data.get('oversight_level', 'N/A')
)
# Load the conversation history
agent.load_conversation(
conversation_history=conversation_history,
total_tokens=log_data.get('total_tokens', 0),
prompt_tokens=log_data.get('prompt_tokens', 0),
completion_tokens=log_data.get('completion_tokens', 0)
)
print("Starting interactive session... (type 'exit' to quit)")
while True:
try:
user_input = input("\nYour turn: ")
if user_input.lower() == 'exit':
break
agent.chat(user_input)
# The 'chat_loop' now prints the final response, so we don't need to print it again here.
except KeyboardInterrupt:
print("\nExiting...")
break
# Ask user if they want to save the new conversation
save_choice = input("\nSave the extended conversation to a new log file? (y/n): ").lower()
if save_choice == 'y':
agent.save_logs(output_dir="interrogation_logs")
print("Session ended.")
if __name__ == "__main__":
main()
|
