fix bugs

Author: seanzhangkx8

…vo_similarity/examples/scd_example.ipynb …/convo_similarity/examples/example.ipynbconvokit/convo_similarity/examples/scd_example.ipynb renamed to convokit/convo_similarity/examples/example.ipynb convokit/convo_similarity/examples/scd_example.ipynb renamed to convokit/convo_similarity/examples/example.ipynb

@@ -55,8 +55,7 @@
    "outputs": [],
    "source": [
     "### Define your own formatter function for your data\n",
-    "def format_friends_transcript_from_convokit(corpus, convo_id):\n",
-    "    convo = corpus.get_conversation(convo_id)\n",
+    "def format_friends_transcript_from_convokit(convo):\n",
     "    utt_lst = convo.get_utterance_ids()\n",
     "    speaker_ids = {}\n",
     "    transcript = \"\"\n",

convokit/convo_similarity/summary.py

@@ -201,7 +201,7 @@ def get_scd_summary(self, transcript: str) -> str:
         :return: SCD summary text
         """
         query = self.SUMMARY_PROMPT_TEMPLATE.format(transcript=transcript)
-        response = self.client.generate(query, model=self.model)
+        response = self.client.generate(query)
         return response.text
 
     def get_sop_from_summary(self, summary: str) -> dict:
@@ -215,7 +215,7 @@ def get_sop_from_summary(self, summary: str) -> dict:
         :raises Exception: If the response cannot be parsed as a dictionary
         """
         query = self.BULLETPOINT_PROMPT_TEMPLATE + summary
-        response = self.client.generate(query, model=self.model)
+        response = self.client.generate(query)
 
         parsed_text = response.text.strip()
 

convokit/genai/gemini_client.py

@@ -26,7 +26,8 @@ def __init__(
         google_cloud_location: str = None,
         use_vertex_ai: bool = True,
     ):
-        os.environ["GOOGLE_API_KEY"] = api_key
+        if not use_vertex_ai:
+            os.environ["GOOGLE_API_KEY"] = api_key
         if use_vertex_ai:
             os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "true"
         if google_cloud_location: