Cast response to correct type

Author: thekaranacharya

guardrails/applications/text2sql.py

@@ -4,6 +4,7 @@
 from string import Template
 from typing import Callable, Dict, Optional, Type, cast
 
+from guardrails.classes import OT, ValidationOutcome
 from guardrails.document_store import DocumentStoreBase, EphemeralDocumentStore
 from guardrails.embedding import EmbeddingBase, OpenAIEmbedding
 from guardrails.guard import Guard
@@ -199,6 +200,7 @@ def __call__(self, text: str) -> Optional[str]:
                     },
                     **self.llm_api_kwargs,
                 )
+                response = cast(ValidationOutcome[OT], response)
                 validated_output: Dict = cast(Dict, response.validated_output)
                 output = validated_output["generated_sql"]
             except TypeError: