#!/usr/bin/env python3
import asyncio
import sys
from gitlab_api.gitlab_agent import stream_chat, chat, node_chat
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
try:
from gitlab_api.gitlab_agent import create_agent
except ImportError as e:
print(f"Import Error: {e}")
print("Please install dependencies via `pip install .[all]`")
sys.exit(1)
async def main():
print("Initializing GitlabA2A Agent...")
try:
agent = create_agent(
provider="openai",
model_id=os.getenv("MODEL_ID", "qwen/qwen3-coder-next"),
base_url=os.getenv("LLM_BASE_URL", "http://host.docker.internal:1234/v1"),
api_key=os.getenv("LLM_API_KEY", "ollama"),
mcp_url=os.getenv("MCP_URL", "http://localhost:8005/mcp"),
mcp_config=None,
)
print("Agent initialized successfully.")
questions = [
"Can you create a merge request for project id 171 with a title of 'Test Merge Request' from the 'validate' to 'main' branch? For the description please put 'this is a test merge request'.",
"Can you run the pipeline for project id 171 on the 'main' branch?",
"Can you create a branch called 'test' from the 'main' branch for project id 171?",
]
print("\n--- Starting Sample Chat Validation ---\n")
for q in questions:
print(f"\n\n\nUser: {q}")
try:
await stream_chat(agent=agent, prompt=q)
await chat(agent=agent, prompt=q)
await node_chat(agent=agent, prompt=q)
if hasattr(agent, "tools"):
print(f"Agent Tools: {[t.__name__ for t in agent.tools]}")
elif hasattr(agent, "_tools"):
print(f"Agent Tools: {[t.__name__ for t in agent._tools]}")
except Exception as e:
print(f"\n\nError processing question '{q}': {e}")
except Exception as e:
print(f"Validation failed with error: {e}")
import traceback
traceback.print_exc()
if __name__ == "__main__":
asyncio.run(main())
