Merge pull request #7 from Zipstack/ssl-verification-arg

Added an argument to disable SSL certificate verification

Author: ritwik-g

README.md

@@ -61,6 +61,7 @@ This will display detailed usage information.
 - `--log_level`: Log level (default: `INFO`).
 - `--print_report`: Print a detailed report of all processed files at the end.
 - `--include_metadata`: Include metadata on tokens consumed and the context passed to LLMs for prompt studio exported tools in the result for each file.
+- `--no_verify`: Disable SSL certificate verification. (By default, SSL verification is enabled.)
 
 ## Usage Examples
 
@@ -141,4 +142,4 @@ On Slack, [join great conversations](https://join-slack.unstract.com/) around LL
 
 [Unstract Cloud](https://unstract.com/): Signup and Try!.
 
-[Unstract developer documentation](https://docs.unstract.com/): Learn more about Unstract and its API.
+[Unstract developer documentation](https://docs.unstract.com/): Learn more about Unstract and its API.

main.py

@@ -35,6 +35,7 @@ class Arguments:
     log_level: str = "INFO"
     print_report: bool = False
     include_metadata: bool = False
+    verify: bool = True
 
 
 # Initialize SQLite DB
@@ -246,6 +247,7 @@ def process_file(
             api_timeout=args.api_timeout,
             logging_level=args.log_level,
             include_metadata=args.include_metadata,
+            verify=args.verify,
         )
 
         status_endpoint, execution_status, response = get_status_endpoint(
@@ -418,6 +420,14 @@ def main():
         help="Include metadata on tokens consumed and the context passed to LLMs for prompt studio exported tools in the result for each file.",
     )
 
+    parser.add_argument(
+        "--no_verify",
+        dest="verify",
+        action="store_false",
+        help="Disable SSL certificate verification.",
+    )
+
+
     args = Arguments(**vars(parser.parse_args()))
 
     ch = logging.StreamHandler(sys.stdout)