Update selfapi_writer.py

selfapi_writer.py

@@ -17,6 +17,7 @@ class SelfApiWriter:
         if not ANTHROPIC_API_KEY:
             raise ValueError("Anthropic API key not found in HuggingFace secrets. Please ensure ANTHROPIC_API_KEY is set in your space's secrets.")
         self.client = Anthropic(api_key=ANTHROPIC_API_KEY)
+        self.model = "claude-3-opus-20240229"
         self.context = {}
         self.book_structure = None
         self.writing_guidelines = None
@@ -69,7 +70,8 @@ class SelfApiWriter:
             Return only the JSON structure without any additional text."""
 
             response = self.client.messages.create(
-                model="claude-3-sonnet-20240229",
+                #model="claude-3-sonnet-20240229",
+                model=self.model,
                 max_tokens=2000,
                 temperature=0,
                 system=system_prompt,
@@ -117,7 +119,7 @@ class SelfApiWriter:
             Follow ALL provided guidelines for style, tone, and content."""
 
             response = self.client.messages.create(
-                model="claude-3-sonnet-20240229",
+                model=self.model,
                 max_tokens=4000,
                 temperature=0.7,
                 system=system_prompt,
@@ -171,7 +173,7 @@ class SelfApiWriter:
             Begin writing the complete chapter now."""
 
             response = self.client.messages.create(
-                model="claude-3-sonnet-20240229",
+                model=self.model,
                 max_tokens=4000,
                 temperature=0.7,
                 system=system_prompt,