Update generate_docs.py

Author: jzielke84

scripts/generate_docs.py

@@ -1,5 +1,5 @@
 import os
-import cohere
+import requests
 
 # -----------------------------
 # Path Setup (Local + GitHub CI)
@@ -15,13 +15,17 @@
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 # -----------------------------
-# Cohere Chat API Setup
+# OpenRouter API Setup
 # -----------------------------
-COHERE_API_KEY = os.getenv("COHERE_API_KEY")
-if not COHERE_API_KEY:
-    raise RuntimeError("COHERE_API_KEY environment variable not set")
+API_KEY = os.getenv("OPENROUTER_API_KEY")
+if not API_KEY:
+    raise RuntimeError("OPENROUTER_API_KEY environment variable not set")
 
-co = cohere.Client(COHERE_API_KEY)
+URL = "https://openrouter.ai/api/v1/chat/completions"
+HEADERS = {
+    "Authorization": f"Bearer {API_KEY}",
+    "Content-Type": "application/json"
+}
 
 # -----------------------------
 # Functions
@@ -39,27 +43,29 @@ def read_ansible_role(role_path):
     return data
 
 def generate_docs(content, role_name):
-    """Generate Markdown documentation using Cohere Chat API"""
-    messages = [
-        {
-            "role": "system",
-            "content": "You are a helpful assistant that generates documentation for Ansible roles in Markdown format."
-        },
-        {
-            "role": "user",
-            "content": f"Document the Ansible role '{role_name}'. Include:\n- Purpose of the role\n- Main tasks\n- Variables and defaults\n- Dependencies or role requirements\n\nYAML content:\n{content}"
-        }
-    ]
+    """Generate Markdown documentation using OpenRouter Chat API"""
+    prompt = f"""
+Document the Ansible role '{role_name}'. Include:
+- Purpose of the role
+- Main tasks
+- Variables and defaults
+- Dependencies or role requirements
 
-    response = co.chat(
-        model="command-a-03-2025",  # Free-tier compatible chat model
-        messages=messages,
-        max_tokens=1000,
-        temperature=0.5
-    )
+YAML content:
+{content}
+"""
 
-    # Access the generated text
-    return response.message.content[0].text.strip()
+    payload = {
+        "model": "gpt-4o-mini",  # free-tier compatible model
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.5,
+        "max_tokens": 1000
+    }
+
+    response = requests.post(URL, headers=HEADERS, json=payload)
+    response.raise_for_status()
+    result = response.json()
+    return result["choices"][0]["message"]["content"].strip()
 
 # -----------------------------
 # Main Script