From cda99f80d259fdec7040e335f779b5cfd490652a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= <hoenicke.florian@gmail.com>
Date: Fri, 19 May 2023 16:45:28 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=8E=20feat:=20search=20fix=20web=20sea?=
 =?UTF-8?q?rch?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dev_gpt/options/generate/templates_user.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dev_gpt/options/generate/templates_user.py b/dev_gpt/options/generate/templates_user.py
index ab36ada..5b61064 100644
--- a/dev_gpt/options/generate/templates_user.py
+++ b/dev_gpt/options/generate/templates_user.py
@@ -126,6 +126,7 @@ image_url_list = search_images('<search term>', top_n=10)
 ```
 """
 
+linebreak = '\n'
 def template_generate_function_constructor(is_using_gpt_3_5_turbo, is_using_google_custom_search):
     return PromptTemplate.from_template(
     general_guidelines_string + f'''
@@ -139,8 +140,7 @@ It will be tested with the following scenario: '{{test_description}}'.
 For the implementation use the following package(s): '{{packages}}'.
 
 The code must start with the following imports:
-```
-from .apis import GPT_3_5_Turbo
+```{linebreak +'from .apis import GPT_3_5_Turbo' if is_using_gpt_3_5_turbo else ""}
 import json
 import requests
 ```