From 5b1ea928540a34ad851a04ae29923edfa503f510 Mon Sep 17 00:00:00 2001
From: Andrew Wang <andrewwa@nvidia.com>
Date: Mon, 18 Nov 2024 17:34:47 -0800
Subject: [PATCH] minor path fixes

---
 services/AgentService/main.py   | 3 ++-
 shared/shared/api_types.py      | 2 +-
 shared/shared/pdf_types.py      | 5 +++--
 shared/shared/prompt_tracker.py | 2 +-
 4 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/services/AgentService/main.py b/services/AgentService/main.py
index 8ec74bf..48de659 100644
--- a/services/AgentService/main.py
+++ b/services/AgentService/main.py
@@ -3,7 +3,8 @@
     ServiceType,
     JobStatus,
 )
-from shared.podcast_types import Conversation, TranscriptionRequest, PodcastOutline
+from shared.podcast_types import Conversation, PodcastOutline
+from shared.api_types import TranscriptionRequest
 from podcast_flow import (
     podcast_summarize_pdfs,
     podcast_generate_raw_outline,
diff --git a/shared/shared/api_types.py b/shared/shared/api_types.py
index 0a1b627..8bb6197 100644
--- a/shared/shared/api_types.py
+++ b/shared/shared/api_types.py
@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field, model_validator
 from typing import Optional, Dict, List
-from pdf_types import PDFMetadata
+from .pdf_types import PDFMetadata
 from enum import Enum
 
 
diff --git a/shared/shared/pdf_types.py b/shared/shared/pdf_types.py
index 565ab2c..a3aadc2 100644
--- a/shared/shared/pdf_types.py
+++ b/shared/shared/pdf_types.py
@@ -1,4 +1,5 @@
 from pydantic import BaseModel, Field
+from typing import Optional
 from datetime import datetime
 from enum import Enum
 
@@ -12,7 +13,7 @@ class PDFConversionResult(BaseModel):
     filename: str
     content: str = ""
     status: ConversionStatus
-    error: str | None = None
+    error: Optional[str] = None
 
 
 class PDFMetadata(BaseModel):
@@ -20,5 +21,5 @@ class PDFMetadata(BaseModel):
     markdown: str = ""
     summary: str = ""
     status: ConversionStatus
-    error: str | None = None
+    error: Optional[str] = None
     created_at: datetime = Field(default_factory=datetime.utcnow)
diff --git a/shared/shared/prompt_tracker.py b/shared/shared/prompt_tracker.py
index 3cc75c8..5c46aef 100644
--- a/shared/shared/prompt_tracker.py
+++ b/shared/shared/prompt_tracker.py
@@ -2,7 +2,7 @@
 import time
 import logging
 from .storage import StorageManager
-from .shared_types import ProcessingStep, PromptTracker as PromptTrackerModel
+from .prompt_types import ProcessingStep, PromptTracker as PromptTrackerModel
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)