Fixing

Author: xisen-w

.DS_Store

@@ -77,7 +77,10 @@ def analyze(self, startup_info, mode):
 
             nl_advanced_analysis = self.get_response(prompt, "Formulate a professional and comprehensive analysis please.")
             self.logger.info("Natural language analysis completed")
-            return nl_advanced_analysis
+            return {
+                'analysis': nl_advanced_analysis,
+                'external_report': external_knowledge
+            }
 
         return analysis
 

agents/market_agent.py

@@ -57,7 +57,10 @@ def analyze(self, startup_info, mode):
 
             nl_advanced_analysis = self.get_response(prompt, "Write a comprehensive report about the product analysis from the VC perspective.")
             self.logger.info("Natural language analysis completed")
-            return nl_advanced_analysis
+            return {
+                'analysis': nl_advanced_analysis,
+                'external_report': product_report
+            }
 
         if mode == "advanced":
             self.logger.info("Starting advanced analysis with external research")

agents/product_agent.py

@@ -2,10 +2,10 @@
 import sys
 import logging
 from pydantic import BaseModel, Field
-from utils.api_wrapper import get_structured_output
+from dotenv import load_dotenv
 
 # Add the project root directory to the Python path
-project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+project_root = os.path.abspath(os.path.join(os.path.dirname(__file__)))
 sys.path.insert(0, project_root)
 
 from agents.base_agent import BaseAgent
@@ -15,15 +15,20 @@ class BaselineAnalysis(BaseModel):
     score: float = Field(..., description="Overall score between 1 and 10")
     recommendation: str = Field(..., description="Recommendation: 'Successful' or 'Unsuccessful'")
 
+# Load environment variables
+load_dotenv()
+
 class BaselineFramework(BaseAgent):
     def __init__(self, model="gpt-4o-mini"):
         super().__init__(model)
         self.logger = logging.getLogger(__name__)
 
-    def analyze_startup(self, startup_info_str: str) -> dict:
+    def analyze_startup(self, startup_info: str) -> dict:
         """Simple baseline analysis using only ChatGPT"""
         self.logger.info("Starting baseline analysis")
 
+        # Format startup info similar to other agents
+        
         prompt = """
         You are an experienced venture capitalist analyzing a startup. Based on the provided information, 
         give a comprehensive analysis and predict if the startup will be successful or not.
@@ -37,41 +42,30 @@ def analyze_startup(self, startup_info_str: str) -> dict:
         """
 
         try:
-            response = self.get_structured_output(BaselineAnalysis, prompt, startup_info_str)
+            response = self.get_json_response(BaselineAnalysis, prompt, "Startup Info: " + startup_info)
             return response.dict()
 
         except Exception as e:
             self.logger.error(f"Error in baseline analysis: {str(e)}")
             return {"error": str(e)}
-        
+
 if __name__ == "__main__":
     def test_baseline_framework():
         # Create a BaselineFramework instance
         framework = BaselineFramework()
 
-        # Test startup info
-        test_startup = """
-        Company Name: TechStart
-        Description: AI-powered software development platform
-        Product Details: Cloud-based IDE with AI assistance for code generation and debugging
-        Technology Stack: Python, React, AWS
-        Founder Background: Ex-Google engineer with 10 years experience in developer tools
-        """
+        # Test startup info following the pattern from FounderAgent test
+        startup_info = "We are a startup that provides a platform for AI-powered software development. Our founders are from Oxford university."
 
         try:
-            # Test analyze_startup method
             print("Testing BaselineFramework analyze_startup:")
             print("-" * 50)
-            result = framework.analyze_startup(test_startup)
+            result = framework.analyze_startup(startup_info)
 
-            # Print results in a formatted way
+            print("\nAnalysis Results:")
             for key, value in result.items():
                 print(f"\n{key}:")
-                if isinstance(value, dict):
-                    for k, v in value.items():
-                        print(f"  {k}: {v}")
-                else:
-                    print(f"  {value}")
+                print(f"  {value}")
 
             print("\nTest completed successfully!")