88
99from magicalapi .types .base import ErrorResponse
1010from magicalapi .types .company_data import CompanyDataResponse
11- from magicalapi .types .schemas import HttpResponse
11+ from magicalapi .types .schemas import HttpResponse , WebhookCreatedResponse
1212
1313from .base_service import BaseService
1414
@@ -23,13 +23,18 @@ async def get_company_data(
2323 company_username : str | None = None ,
2424 company_name : str | None = None ,
2525 company_website : str | None = None ,
26- ) -> CompanyDataResponse | ErrorResponse :
26+ ) -> CompanyDataResponse | WebhookCreatedResponse | ErrorResponse :
2727 """this method sends request to company data service in magicalAPI.
2828 https://magicalapi.com/services/company-data
2929
3030 company_name (``str``):
3131 the username of linkedin company that you want to get it's data.
3232
33+ Returns:
34+ CompanyDataResponse: When request completes successfully (no webhook).
35+ WebhookCreatedResponse: When using webhook_url (immediate acknowledgment).
36+ ErrorResponse: When an error occurs (e.g., 403 if webhook domain not whitelisted).
37+
3338 """
3439 # check which parameters passed
3540 if company_username is not None :
@@ -66,5 +71,5 @@ async def get_company_data(
6671
6772 def validate_response (
6873 self , response : HttpResponse , validate_model : type [BaseModel ]
69- ) -> CompanyDataResponse | ErrorResponse :
74+ ) -> CompanyDataResponse | WebhookCreatedResponse | ErrorResponse :
7075 return super ().validate_response (response , validate_model )
0 commit comments