Skip to content

Commit 3e49c6a

Browse files
authored
Merge pull request #1684 from swirlai/ds-4629-develop
cherry-pick from main, removing temperature parameter for ChatGPT-5 …
2 parents a36f4af + aafa71e commit 3e49c6a

File tree

4 files changed

+17
-22
lines changed

4 files changed

+17
-22
lines changed

swirl/connectors/gen_ai.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,8 +79,7 @@ def execute_search(self, session=None):
7979
messages=[
8080
{"role": "system", "content": self.system_guide},
8181
{"role": "user", "content": self.query_to_provider},
82-
],
83-
temperature=0
82+
]
8483
)
8584
message = completions.choices[0].message.content
8685
self.found = 1

swirl/processors/gen_ai_query.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,16 +106,14 @@ def process(self, client=None):
106106
messages=[
107107
{"role": "system", "content": self.system_guide},
108108
{"role": "user", "content": self.prompt.format(query_string=self.query_string) },
109-
],
110-
temperature=0
109+
]
111110
)
112111
response = client.openai_client.chat.completions.create(
113112
model=client.get_model(),
114113
messages=[
115114
{"role": "system", "content": self.system_guide},
116115
{"role": "user", "content": self.prompt.format(query_string=self.query_string) },
117-
],
118-
temperature=0
116+
]
119117
)
120118
message = response.choices[0].message.content
121119
logger.info(f"ChatGPT Response: {message}")

swirl/processors/rag.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -232,8 +232,7 @@ def background_process(self):
232232
messages=[
233233
{"role": "system", "content": rag_prompt.get_role_system_guide_text()},
234234
{"role": "user", "content": new_prompt_text},
235-
],
236-
temperature=0
235+
]
237236
)
238237
model_response = completions_new.choices[0].message.content
239238
logger.warning(f'RAG: fetch_prompt_errors follow:')

swirl/tests/tests.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -601,8 +601,7 @@ def test_cgptqp_1():
601601
messages=[
602602
{"role": "system", "content": "You are helping a user formulate better queries"},
603603
{"role": "user", "content": "Write a more precise query of similar length to this : gig economy"}
604-
],
605-
temperature=0
604+
]
606605
)
607606

608607
@pytest.mark.django_db
@@ -630,8 +629,8 @@ def test_cgptqp_2():
630629
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
631630
{"role": "system", "content": "You are a malevolent dictator"},
632631
{"role": "user", "content": "Write a more precise query of similar length to this : gig economy"}
633-
],
634-
temperature=0)
632+
]
633+
)
635634

636635
@pytest.mark.django_db
637636
def test_cgptqp_3():
@@ -658,8 +657,8 @@ def test_cgptqp_3():
658657
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
659658
{"role": "system", "content": "You are a malevolent dictator"},
660659
{"role": "user", "content": "Write a more precise query of similar length to this : gig economy"}
661-
],
662-
temperature=0)
660+
]
661+
)
663662

664663
@pytest.mark.django_db
665664
def test_cgptqp_4():
@@ -688,8 +687,8 @@ def test_cgptqp_4():
688687
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
689688
{"role": "system", "content": "You are a malevolent dictator"},
690689
{"role": "user", "content": "This should be used: gig economy"}
691-
],
692-
temperature=0)
690+
]
691+
)
693692

694693

695694
@pytest.mark.django_db
@@ -721,8 +720,8 @@ def test_cgptqp_5():
721720
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
722721
{"role": "system", "content": "You are a malevolent dictator"},
723722
{"role": "user", "content": "This should be used: gig economy"}
724-
],
725-
temperature=0)
723+
]
724+
)
726725

727726

728727
@pytest.mark.django_db
@@ -754,8 +753,8 @@ def test_cgptqp_6():
754753
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
755754
{"role": "system", "content": "You are a malevolent dictator"},
756755
{"role": "user", "content": "This should be used: gig economy"}
757-
],
758-
temperature=0)
756+
]
757+
)
759758

760759

761760
@pytest.mark.django_db
@@ -787,8 +786,8 @@ def test_cgptqp_7():
787786
mock_create.assert_called_once_with(model=TEST_AI_MODEL, messages=[
788787
{"role": "system", "content": "You are a malevolent dictator"},
789788
{"role": "user", "content": "This should be used: gig economy"}
790-
],
791-
temperature=0)
789+
]
790+
)
792791

793792

794793
@pytest.fixture

0 commit comments

Comments
 (0)