| from core.client import BeamClient | |
| from scenarios import SCENARIO_01_AUTH | |
| client = BeamClient() | |
| prompt = f"""You are an autonomous software engineer. | |
| Fix the bugs in auth.py so that tests/test_auth.py passes. | |
| CURRENT FILES: | |
| auth.py: | |
| {SCENARIO_01_AUTH.files['auth.py']} | |
| tests/test_auth.py: | |
| {SCENARIO_01_AUTH.files['tests/test_auth.py']} | |
| Provide the fixed auth.py inside <edit_file path="auth.py">...</edit_file>. | |
| """ | |
| print("Testing STREAMING with Beam...") | |
| reasoning_chunks = [] | |
| content_chunks = [] | |
| resp = client.stream_chat( | |
| messages=[{"role": "user", "content": prompt}], | |
| max_tokens=16000, | |
| on_reasoning=lambda r: reasoning_chunks.append(r), | |
| on_content=lambda c: content_chunks.append(c) | |
| ) | |
| print(f"\nReasoning Chunks: {len(reasoning_chunks)} (Length: {len(''.join(reasoning_chunks))})") | |
| print(f"Content Chunks: {len(content_chunks)} (Length: {len(''.join(content_chunks))})") | |
| print("\n--- CONTENT GENERATED ---") | |
| print("".join(content_chunks)) | |