diagram-to-iac 0.1.0__py3-none-any.whl

diagram_to_iac/cli.py

@@ -0,0 +1,13 @@
+# src/diagram_to_iac/cli.py
+
+import sys, json
+import requests   # external package, so pipreqs must pick it up
+
+def main():
+    # a sanity check that our imports actually work:
+    print("✅  diagram-to-iac CLI is up and running!")
+    print("• requests version:", requests.__version__)
+    # verify stdlib too
+    data = {"ok": True}
+    print("• json dump:", json.dumps(data))
+    sys.exit(0)

diagram_to_iac/tools/api_utils.py

@@ -0,0 +1,109 @@
+import sys
+import os
+from openai import OpenAI
+from anthropic import Anthropic
+import requests
+import google.generativeai as genai
+import googleapiclient.discovery
+# Add the parent directory to sys.path to import env_loader
+# sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+# from scripts.env_loader import load_env_keys
+
+def test_openai_api():
+    try:
+        if not os.environ.get("OPENAI_API_KEY"):
+            print("❌ OpenAI API error: OPENAI_API_KEY environment variable not set.")
+            return False
+        client = OpenAI()
+        response = client.chat.completions.create(
+            # Consider using a newer model if appropriate, e.g., "gpt-4o-mini" or "gpt-3.5-turbo"
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": "Hello, are you working?"}],
+            max_tokens=10
+        )
+        return True
+    except Exception as e:
+        print(f"❌ Open AI API error: {str(e)}")
+        return False
+
+def test_gemini_api():
+    try:
+        google_api_key = os.environ.get("GOOGLE_API_KEY")
+        if not google_api_key:
+            print("❌ Gemini API error: GOOGLE_API_KEY environment variable not set.")
+            return False
+        genai.configure(api_key=google_api_key)
+        model = genai.GenerativeModel('gemini-1.5-flash-latest') # Corrected model name
+        response = model.generate_content("Hello, are you working?")
+        return True
+    except Exception as e:
+        print(f"❌ Gemini API error: {str(e)}")
+        return False
+
+def test_anthropic_api():
+    try:
+        if not os.environ.get("ANTHROPIC_API_KEY"):
+            print("❌ Anthropic API error: ANTHROPIC_API_KEY environment variable not set.")
+            return False
+        client = Anthropic()
+        response = client.messages.create(
+            model="claude-3-haiku-20240307",
+            max_tokens=10,
+            messages=[{"role": "user", "content": "Hello, are you working?"}]
+        )
+        return True
+    except Exception as e:
+        print(f"❌ Anthropic API error: {str(e)}")
+        return False
+
+def test_github_api():
+    """Test the GitHub API connection."""
+    try:      
+        token = os.environ.get("GITHUB_TOKEN")
+        if not token: # This check is already good
+            print("❌ GitHub API error: GITHUB_TOKEN environment variable not set")
+            return False
+            
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "Accept": "application/vnd.github.v3+json"
+        }
+        
+        # Try to get the authenticated user
+        url = "https://api.github.com/user"
+        response = requests.get(url, headers=headers)
+        
+        if response.status_code == 200:
+            user_data = response.json()
+            username = user_data.get('login')
+            print(f"✅ GitHub API works! Authenticated as: {username}")
+            return True
+        else:
+            print(f"❌ GitHub API error: Status code {response.status_code}")
+            return False
+    except Exception as e:
+        print(f"❌ GitHub API error: {str(e)}")
+        return False
+
+def test_all_apis():
+    print("Hello from the test workflow!")
+    print("Loading environment keys...")
+    # Load environment keys once before running all tests
+    #load_env_keys()
+    
+    print("Testing API connections...")
+    openai_success = test_openai_api()
+    gemini_success = test_gemini_api()
+    anthropic_success = test_anthropic_api()
+    github_success = test_github_api()  # Added GitHub API test
+    
+    print("\nSummary:")
+    print(f"OpenAI API: {'✅ Working' if openai_success else '❌ Failed'}")
+    print(f"Gemini API: {'✅ Working' if gemini_success else '❌ Failed'}")
+    print(f"Anthropic API: {'✅ Working' if anthropic_success else '❌ Failed'}")
+    print(f"GitHub API: {'✅ Working' if github_success else '❌ Failed'}")  # Added GitHub API result
+    
+    if openai_success and gemini_success and anthropic_success and github_success:  # Updated condition
+        print("\n🎉 All APIs are working correctly!")
+    else:
+        print("\n⚠️ Some APIs failed. Check the errors above.")

diagram_to_iac/tools/sec_utils.py

@@ -0,0 +1,64 @@
+# src/diagram_to_iac/tools/secrets.py
+
+"""
+Decode /run/secrets.yaml (Base64‑encoded values) into real env vars.
+
+Mount it into the container with:
+  docker run … -v "$PWD/config/secrets.yaml":/run/secrets.yaml:ro …
+
+If the file is missing (e.g. CI injects GitHub Secrets directly),
+load_yaml_secrets() is a no‑op.
+"""
+
+import os
+import base64
+import yaml
+import pathlib
+import binascii
+
+# Path inside container where the encoded YAML is mounted
+_YAML_PATH = pathlib.Path("/run/secrets.yaml")
+
+
+def _decode_b64(enc: str) -> str:
+    """Robust Base64 decode: fixes padding, falls back if invalid."""
+    enc = enc.strip()
+    if not enc:
+        return ""
+    # Fix missing padding
+    enc += "=" * (-len(enc) % 4)
+    try:
+        return base64.b64decode(enc).decode("utf-8").strip()
+    except (binascii.Error, UnicodeDecodeError):
+        # If it isn’t valid Base64, return the raw string
+        return enc
+
+
+def load_yaml_secrets() -> None:
+    """
+    Read /run/secrets.yaml, decode each *_ENCODED value, and export
+    as environment variables. Special‑case REPO_TOKEN → GITHUB_TOKEN.
+    Safe to call when the file does not exist.
+    """
+    if not _YAML_PATH.exists():
+        return
+
+    data: dict[str, str] = yaml.safe_load(_YAML_PATH.read_text()) or {}
+    for key, encoded in data.items():
+        if not encoded:
+            continue
+
+        # Strip the "_ENCODED" suffix
+        # base_name = key.removesuffix("_ENCODED")
+
+        # GitHub Actions requires the env var "GITHUB_TOKEN"
+        # not "REPO_TOKEN", so map accordingly
+        if key == "REPO_TOKEN":
+            env_name = "GITHUB_TOKEN"
+        else:
+            env_name = key
+
+        # Decode and export
+        plain_value = _decode_b64(str(encoded))
+        os.environ[env_name] = plain_value
+        # print(f"Decoded {env_name}={plain_value}")

diagram_to_iac-0.1.0.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.4
+Name: diagram-to-iac
+Version: 0.1.0
+Summary: Convert architecture diagrams into IaC modules
+Author-email: vindpro <admin@vindpro.com>
+Description-Content-Type: text/markdown
+Requires-Dist: anthropic==0.51.0
+Requires-Dist: google_api_python_client==2.169.0
+Requires-Dist: openai==1.79.0
+Requires-Dist: protobuf<5.0.0dev,>=3.19.5
+Requires-Dist: PyYAML==6.0.2
+Requires-Dist: Requests==2.32.3
+Requires-Dist: google-generativeai==0.3.1
+
+# diagram-to-iac
+an automated system that intelligently translates visual cloud infrastructure diagrams (specifically Azure) into deployable Infrastructure as Code (IaC), leveraging AI for analysis and GitHub for collaborative refinement.

diagram_to_iac-0.1.0.dist-info/RECORD

@@ -0,0 +1,30 @@
+diagram_to_iac/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/cli.py,sha256=uumG1frF42eCkdLIZxyxQB1x6lDwtG-qKL4vcHnLLXY,400
+diagram_to_iac/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/codegen_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/consensus_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/github_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/interpretation_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/question_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/supervisor.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/agents/vision_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/core/agent_base.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/core/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/core/memory.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/api_utils.py,sha256=Guq0lJZIc-ds46-2Id80lIOFdbN4vh_dEAM9ONSgS6M,4171
+diagram_to_iac/tools/cv_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/gh_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/sec_utils.py,sha256=SJvNk-OWrF-frIVSYiPirnV4m4-ivzFsNrIXz-SVAxM,1845
+diagram_to_iac/tools/tf_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/llm_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/llm_utils/anthropic_driver.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/llm_utils/gemini_driver.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/llm_utils/openai_driver.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac/tools/llm_utils/router.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+diagram_to_iac-0.1.0.dist-info/METADATA,sha256=xqOahnAIgsnWMDfQUlCk3Dn9NZtIa8CaEubCDs9U6sg,693
+diagram_to_iac-0.1.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+diagram_to_iac-0.1.0.dist-info/entry_points.txt,sha256=E_krzrW2-_KGXGPWFW9derAqSIQ7WYSd2pwrb7ap4j8,59
+diagram_to_iac-0.1.0.dist-info/top_level.txt,sha256=k1cV0YODiCUU46qlmbQaquMcbMXhNm05NZLxsinDUBA,15
+diagram_to_iac-0.1.0.dist-info/RECORD,,

diagram_to_iac-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.7.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

diagram_to_iac-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+diagram-to-iac = diagram_to_iac.cli:main

diagram_to_iac-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+diagram_to_iac