feat(sdk): allow input text into the python sdk (#2738)

GitOrigin-RevId: 9c36a4ad2fa7bd831920145ecf1c29b16765cc11

Author: jhazenaai

README.md

@@ -266,6 +266,37 @@ print(result.response)
 
 </details>
 
+
+<details>
+  <summary>Use LeMUR to with Input Text</summary>
+
+```python
+import assemblyai as aai
+
+transcriber = aai.Transcriber()
+config = aai.TranscriptionConfig(
+  speaker_labels=True,
+)
+transcript = transcriber.transcribe("https://example.org/customer.mp3", config=config)
+
+# Example converting speaker label utterances into LeMUR input text
+text = ""
+
+for utt in transcript.utterances:
+    text += f"Speaker {utt.speaker}:\n{utt.text}\n"
+
+result = aai.Lemur().task(
+  "You are a helpful coach. Provide an analysis of the transcript "
+  "and offer areas to improve with exact quotes. Include no preamble. "
+  "Start with an overall summary then get into the examples with feedback.",
+  input_text=text
+)
+
+print(result.response)
+```
+
+</details>
+
 <details>
   <summary>Delete data previously sent to LeMUR</summary>
 

assemblyai/lemur.py

@@ -14,7 +14,11 @@ def __init__(
     ) -> None:
         self._client = client
 
-        self._sources = [types.LemurSourceRequest.from_lemur_source(s) for s in sources]
+        self._sources = (
+            [types.LemurSourceRequest.from_lemur_source(s) for s in sources]
+            if sources is not None
+            else []
+        )
 
     def question(
         self,
@@ -24,6 +28,7 @@ def question(
         final_model: Optional[types.LemurModel],
         max_output_size: Optional[int],
         temperature: Optional[float],
+        input_text: Optional[str],
     ) -> types.LemurQuestionResponse:
         response = api.lemur_question(
             client=self._client.http_client,
@@ -34,6 +39,7 @@ def question(
                 final_model=final_model,
                 max_output_size=max_output_size,
                 temperature=temperature,
+                input_text=input_text,
             ),
             http_timeout=timeout,
         )
@@ -48,6 +54,7 @@ def summarize(
         max_output_size: Optional[int],
         timeout: Optional[float],
         temperature: Optional[float],
+        input_text: Optional[str],
     ) -> types.LemurSummaryResponse:
         response = api.lemur_summarize(
             client=self._client.http_client,
@@ -58,6 +65,7 @@ def summarize(
                 final_model=final_model,
                 max_output_size=max_output_size,
                 temperature=temperature,
+                input_text=input_text,
             ),
             http_timeout=timeout,
         )
@@ -72,6 +80,7 @@ def action_items(
         max_output_size: Optional[int],
         timeout: Optional[float],
         temperature: Optional[float],
+        input_text: Optional[str],
     ) -> types.LemurActionItemsResponse:
         response = api.lemur_action_items(
             client=self._client.http_client,
@@ -82,6 +91,7 @@ def action_items(
                 final_model=final_model,
                 max_output_size=max_output_size,
                 temperature=temperature,
+                input_text=input_text,
             ),
             http_timeout=timeout,
         )
@@ -95,6 +105,7 @@ def task(
         max_output_size: Optional[int],
         timeout: Optional[float],
         temperature: Optional[float],
+        input_text: Optional[str],
     ):
         response = api.lemur_task(
             client=self._client.http_client,
@@ -104,6 +115,7 @@ def task(
                 final_model=final_model,
                 max_output_size=max_output_size,
                 temperature=temperature,
+                input_text=input_text,
             ),
             http_timeout=timeout,
         )
@@ -121,7 +133,7 @@ class Lemur:
 
     def __init__(
         self,
-        sources: List[types.LemurSource],
+        sources: Optional[List[types.LemurSource]] = None,
         client: Optional[_client.Client] = None,
     ) -> None:
         """
@@ -147,6 +159,7 @@ def question(
         max_output_size: Optional[int] = None,
         timeout: Optional[float] = None,
         temperature: Optional[float] = None,
+        input_text: Optional[str] = None,
     ) -> types.LemurQuestionResponse:
         """
         Question & Answer allows you to ask free form questions about one or many transcripts.
@@ -178,6 +191,7 @@ def question(
             max_output_size=max_output_size,
             timeout=timeout,
             temperature=temperature,
+            input_text=input_text,
         )
 
     def summarize(
@@ -188,6 +202,7 @@ def summarize(
         max_output_size: Optional[int] = None,
         timeout: Optional[float] = None,
         temperature: Optional[float] = None,
+        input_text: Optional[str] = None,
     ) -> types.LemurSummaryResponse:
         """
         Summary allows you to distill a piece of audio into a few impactful sentences.
@@ -214,6 +229,7 @@ def summarize(
             max_output_size=max_output_size,
             timeout=timeout,
             temperature=temperature,
+            input_text=input_text,
         )
 
     def action_items(
@@ -224,6 +240,7 @@ def action_items(
         max_output_size: Optional[int] = None,
         timeout: Optional[float] = None,
         temperature: Optional[float] = None,
+        input_text: Optional[str] = None,
     ) -> types.LemurActionItemsResponse:
         """
         Action Items allows you to generate action items from one or many transcripts.
@@ -251,6 +268,7 @@ def action_items(
             max_output_size=max_output_size,
             timeout=timeout,
             temperature=temperature,
+            input_text=input_text,
         )
 
     def task(
@@ -260,6 +278,7 @@ def task(
         max_output_size: Optional[int] = None,
         timeout: Optional[float] = None,
         temperature: Optional[float] = None,
+        input_text: Optional[str] = None,
     ) -> types.LemurTaskResponse:
         """
         Task feature allows you to submit a custom prompt to the model.
@@ -282,6 +301,7 @@ def task(
             max_output_size=max_output_size,
             timeout=timeout,
             temperature=temperature,
+            input_text=input_text,
         )
 
     @classmethod

assemblyai/types.py

@@ -1773,6 +1773,7 @@ class BaseLemurRequest(BaseModel):
     final_model: Optional[LemurModel]
     max_output_size: Optional[int]
     temperature: Optional[float]
+    input_text: Optional[str]
 
 
 class LemurTaskRequest(BaseLemurRequest):