File size: 1,628 Bytes
8e06e17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1c92419
8e06e17
 
 
1c92419
 
 
 
 
 
 
 
 
8e06e17
1c92419
412b084
 
 
1c92419
 
8e06e17
1c92419
 
 
8e06e17
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
from workflowschema import LibraryModel
import httpx
import asyncio
import yaml


class WorkflowInvocation:
    def __init__(self, payload):
        self.config = self.load_config()
        self.payload = payload
        
    def load_config(self):
        with open("config.yaml","r") as file:
            config = yaml.safe_load(file)
        return config

    
    async def invoke_workflow(self): 
        try:   
            webHookUrl = self.config['workflow']['webhook_url']   
            #print(webHookUrl)
            async with httpx.AsyncClient() as client:
                response = await client.post(webHookUrl, json=self.payload)
                response.raise_for_status()
                try:
                # Try to parse JSON response
                    json_response = response.json()
                    print("JSON Response:", json_response)
                    return json_response
                except Exception as e:
                # Fall back to raw text
                    print("Non-JSON response:", response.text)
                    return response.text               
                #print("Response status code:", response.json())
        except httpx.RequestError as e:
            print(f"Request error occurred: {type(e).__name__}: {str(e)}")
            import traceback
            traceback.print_exc()
        except httpx.HTTPStatusError as e:
            print(f"HTTP error occurred: {e.response.status_code} - {e.response.text}")
        except Exception as e:
            import traceback
            print("Unexpected error:", str(e))
            traceback.print_exc()